fixes

NN/models/cob_rl_model.py

@@ -229,8 +229,8 @@ class COBRLModelInterface(ModelInterface):
     Interface for the COB RL model that handles model management, training, and inference
     """
     
-    def __init__(self, model_checkpoint_dir: str = "models/realtime_rl_cob", device: str = None):
-        super().__init__(name="cob_rl_model")  # Initialize ModelInterface with a name
+    def __init__(self, model_checkpoint_dir: str = "models/realtime_rl_cob", device: str = None, name=None, **kwargs):
+        super().__init__(name=name)  # Initialize ModelInterface with a name
         self.model_checkpoint_dir = model_checkpoint_dir
         self.device = torch.device(device if device else ('cuda' if torch.cuda.is_available() else 'cpu'))
         

NN/models/dqn_agent.py

@@ -5,7 +5,7 @@ import numpy as np
 from collections import deque
 import random
 from typing import Tuple, List
-import osvu
+import os
 import sys
 import logging
 import torch.nn.functional as F