fixing not implemented error

CAVEconnectome · Oct 16, 2023 · f6d8c7b · f6d8c7b
1 parent a0a6636
commit f6d8c7b
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/dynamicannotationdb/database.py b/dynamicannotationdb/database.py
@@ -99,13 +99,17 @@ def get_unique_string_values(self, table_name: str):
         unique_values = {}
         with self.session_scope() as session:
             for column_name in model.__table__.columns.keys():
-     
+
                 # if the column is a string
-                if model.__table__.columns[column_name].type.python_type == str:
+                try:
+                    python_type = model.__table__.columns[column_name].type.python_type
+                except NotImplementedError:
+                    python_type = None
+                if python_type == str:
                     query = session.query(getattr(model, column_name)).distinct()
                     unique_values[column_name] = [row[0] for row in query.all()]
         return unique_values
-    
+
     def get_views(self, datastack_name: str):
         with self.session_scope() as session:
             query = session.query(AnalysisView).filter(