use update_workspace callback

nanome-ai · Nov 17, 2023 · c82c67a · c82c67a
1 parent 982edff
commit c82c67a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/testing/test_plugins/InteractionTestAsync.py b/testing/test_plugins/InteractionTestAsync.py
@@ -130,8 +130,7 @@ async def setup_test_workspace(self):
         """Create fresh workspace and load it into Nanome."""
         self.pdb_file = os.path.join(test_pdbs, '1tyl.pdb')
         workspace = Workspace()
-        self.update_workspace(workspace)
-        await asyncio.sleep(1)  # Wait for workspace to be loaded. V annoying.
+        await self.update_workspace(workspace)
         await self.send_files_to_load(self.pdb_file)
         ws = await self.request_workspace()
         comp = ws.complexes[0]