update model revisions in testing.py

cognitedata · Jan 19, 2025 · efdbb0f · efdbb0f
1 parent cef35b7
commit efdbb0f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cognite/client/testing.py b/cognite/client/testing.py
@@ -45,7 +45,7 @@
 from cognite.client._api.sequences import SequencesAPI, SequencesDataAPI
 from cognite.client._api.simulators import SimulatorsAPI
 from cognite.client._api.simulators.integrations import SimulatorIntegrationsAPI
-from cognite.client._api.simulators.models import SimulatorModelsAPI
+from cognite.client._api.simulators.models import SimulatorModelRevisionsAPI, SimulatorModelsAPI
 from cognite.client._api.synthetic_time_series import SyntheticDatapointsAPI
 from cognite.client._api.templates import (
     TemplateGroupsAPI,
@@ -148,6 +148,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
         self.simulators = MagicMock(spec=SimulatorsAPI)
         self.simulators.integrations = MagicMock(spec_set=SimulatorIntegrationsAPI)
         self.simulators.models = MagicMock(spec_set=SimulatorModelsAPI)
+        self.simulators.models.revisions = MagicMock(spec_set=SimulatorModelRevisionsAPI)
 
         self.sequences = MagicMock(spec=SequencesAPI)
         self.sequences.rows = MagicMock(spec_set=SequencesDataAPI)