Add pid to stream wrapper

thatguy11325 · Oct 20, 2024 · 73cb1cd · 73cb1cd
1 parent 0aac8b5
commit 73cb1cd
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/pokemonred_puffer/wrappers/stream_wrapper.py b/pokemonred_puffer/wrappers/stream_wrapper.py
@@ -1,5 +1,6 @@
 import asyncio
 import json
+import os
 
 import gymnasium as gym
 import websockets
@@ -14,10 +15,7 @@ def __init__(self, env: RedGymEnv, config: DictConfig):
 
         self.user = config.user
         self.ws_address = "wss://transdimensional.xyz/broadcast"
-        self.stream_metadata = {
-            "user": self.user,
-            "env_id": self.env_id,
-        }
+        self.stream_metadata = {"user": self.user, "env_id": self.env_id, "extra": os.getpid()}
         self.loop = asyncio.new_event_loop()
         asyncio.set_event_loop(self.loop)
         self.websocket = self.loop.run_until_complete(self.establish_wc_connection())