log artifact

thatguy11325 · Jun 21, 2024 · 4c8fffe · 4c8fffe
1 parent 5faada1
commit 4c8fffe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py
@@ -487,7 +487,7 @@ def close(self):
             artifact = wandb.Artifact(artifact_name, type="model")
             model_path = self.save_checkpoint()
             artifact.add_file(model_path)
-            self.wandb_client.run.log_artifact(artifact)
+            self.wandb_client.log_artifact(artifact)
             self.wandb_client.finish()
 
     def save_checkpoint(self):

diff --git a/pokemonred_puffer/wrappers/stream_wrapper.py b/pokemonred_puffer/wrappers/stream_wrapper.py
@@ -32,6 +32,7 @@ def __init__(self, env: RedGymEnv, config: pufferlib.namespace):
 
     def __del__(self):
         try:
+            self.loop.run_until_complete()
             for task in asyncio.all_tasks():
                 task.cancel()
             self.loop.close()