From f39aae509182ac1100182ad0de47910f4ad23bf1 Mon Sep 17 00:00:00 2001
From: mavaylon1 <mavaylon1@berkeley.edu>
Date: Sun, 19 May 2024 21:11:47 -0700
Subject: [PATCH] clean up

---
 src/pynwb/__init__.py | 2 --
 src/pynwb/core.py     | 1 +
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/pynwb/__init__.py b/src/pynwb/__init__.py
index a5be48bd7..4c957a2be 100644
--- a/src/pynwb/__init__.py
+++ b/src/pynwb/__init__.py
@@ -164,8 +164,6 @@ def register_class(**kwargs):
 
     def _dec(cls):
         __TYPE_MAP.register_container_type(namespace, neurodata_type, cls)
-        cls.type_map = __TYPE_MAP
-        cls.namespace = namespace
         return cls
     if container_cls is None:
         return _dec
diff --git a/src/pynwb/core.py b/src/pynwb/core.py
index 32516eed6..f9ae2bd2f 100644
--- a/src/pynwb/core.py
+++ b/src/pynwb/core.py
@@ -55,6 +55,7 @@ def data_type(self):
         """
         Return the spec data type associated with this container, i.e., the neurodata_type.
         """
+        # we need this function here to use the correct _data_type_attr.
         _type = getattr(self, self._data_type_attr)
         return _type