From 1f7c6f10ab36652b4e5e43f0208292dfd92ba114 Mon Sep 17 00:00:00 2001
From: Brandon Rozek <rozekbrandon@gmail.com>
Date: Tue, 14 Apr 2020 14:21:31 -0400
Subject: [PATCH] Used standard library definition

---
 rltorch/log.py | 38 +++++---------------------------------
 1 file changed, 5 insertions(+), 33 deletions(-)

diff --git a/rltorch/log.py b/rltorch/log.py
index d24a08f..ff3335f 100644
--- a/rltorch/log.py
+++ b/rltorch/log.py
@@ -1,42 +1,14 @@
-from collections import Counter
+from collections import Counter, defaultdict
+from typing import Dict, List, Any
 import numpy as np
 import torch
 
-class Logger:
-    """
-    Keeps track of lists of items seperated by tags.
-
-    Notes
-    -----
-    Logger is a dictionary of lists.
-    """
-    def __init__(self):
-        self.log = {}
-    def append(self, tag, value):
-        if tag not in self.log.keys():
-            self.log[tag] = []
-        self.log[tag].append(value)
-    def clear(self):
-        self.log.clear()
-    def keys(self):
-        return self.log.keys()
-    def __len__(self):
-        return len(self.log)
-    def __iter__(self):
-        return iter(self.log)
-    def __contains__(self, value):
-        return value in self.log
-    def __getitem__(self, index):
-        return self.log[index]
-    def __setitem__(self, index, value):
-        self.log[index] = value
-    def __reversed__(self):
-        return reversed(self.log)
+Logger: Dict[Any, List[Any]] = defaultdict(list)
 
 class LogWriter:
     """
-    Takes a logger and writes it to a writter. 
-    While keeping track of the number of times it 
+    Takes a logger and writes it to a writter.
+    While keeping track of the number of times it
     a certain tag.
 
     Notes