Forcing mock to shutdown in test cleanup
@@ -44,6 +44,7 @@ class AbstractCrowdsourcingTest:
torch.manual_seed(0)
self.operator = None
+ self.server = None
def _teardown(self):
"""
@@ -55,6 +56,9 @@ class AbstractCrowdsourcingTest:
if self.operator is not None:
self.operator.force_shutdown()
+ if self.server is not None:
+ self.server.shutdown_mock()
+
def _set_up_config(
self,
blueprint_type: str,
