Different mock method.

dls-controls · Apr 16, 2019 · 4e0832e · 4e0832e
1 parent d27fe39
commit 4e0832e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/conf.py b/docs/conf.py
@@ -28,7 +28,7 @@ class DataSource(object):
 import pytac
 pytac.exceptions.FieldException = Exception
 pytac.exceptions.HandleException = Exception
-sys.modules['pytac.data_source'] = types.ModuleType('data_source')
+pytac.__dict__['data_source'] = types.ModuleType('data_source')
 pytac.data_source.DataSource = DataSource
 
 class Mock(MagicMock):