From 1eef691c495bcf644cec353de3760796e6de0ecb Mon Sep 17 00:00:00 2001
From: ramoj <ramoj@google.com>
Date: Mon, 6 Jan 2025 02:38:42 +0000
Subject: [PATCH] Progress

---
 dftimewolf/lib/containers/manager.py | 5 ++---
 tests/lib/containers/manager.py      | 9 +--------
 2 files changed, 3 insertions(+), 11 deletions(-)

diff --git a/dftimewolf/lib/containers/manager.py b/dftimewolf/lib/containers/manager.py
index 2548b240..b434ca82 100644
--- a/dftimewolf/lib/containers/manager.py
+++ b/dftimewolf/lib/containers/manager.py
@@ -20,7 +20,7 @@ class _MODULE():
 
 
 class ContainerManager():
-  """A ConatinerManager.
+  """A ContainerManager.
   
   This ContainerManager handles container storage and delivery to modules.
   Modules can only receive containers from other modules that they directly
@@ -29,8 +29,7 @@ class ContainerManager():
 
   Attributes:
     _mutex: Practice safe container access.
-    _dependencies: The module dependency mappings.
-    _storage: Container storage.
+    _modules: Container storage and dependency information.
   """
 
   def __init__(self):
diff --git a/tests/lib/containers/manager.py b/tests/lib/containers/manager.py
index 787e13c7..ab336e8d 100644
--- a/tests/lib/containers/manager.py
+++ b/tests/lib/containers/manager.py
@@ -79,17 +79,10 @@ def __eq__(self, other: "_TestContainer1"):
     return self.param == other.param
 
 
-class _TestContainer2(interface.AttributeContainer):
+class _TestContainer2(_TestContainer1):
   """A Test container."""
   CONTAINER_TYPE = 'test2'
 
-  def __init__(self, param: str):
-    super().__init__()
-    self.param = param
-
-  def __eq__(self, other: "_TestContainer2"):
-    return self.param == other.param
-
 
 class ContainerManagerTest(unittest.TestCase):
   """Tests for the ContainerManager."""