Stick with what works.

dls-controls · Apr 16, 2019 · 393ff4e · 393ff4e
1 parent a511d57
commit 393ff4e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/docs/conf.py b/docs/conf.py
@@ -21,14 +21,17 @@
 # -- Mock modules ------------------------------------------------------------
 
 # If we cannot import pytac prooperly or at then partially or fully mock them. # This is a temporary fix because the latest versions are not yet on pypi.
+class DataSource(object):
+    pass
+
 import pytac
 try:
     pytac.data_source
 except AttributeError:
     pytac.exceptions.FieldException = Exception
     pytac.exceptions.HandleException = Exception
     pytac.__dict__['data_source'] = types.ModuleType('data_source')
-    pytac.data_source.DataSource = object()
+    pytac.data_source.DataSource = DataSource
 
 class Mock(MagicMock):
     @classmethod