Merge commit 'd49a473ed6f4aef55bfdd47d6370e53582be6b7b' into cleanup

NN/models/cob_rl_model.py

@@ -267,7 +267,17 @@ class COBRLModelInterface(ModelInterface):
         
         logger.info(f"COB RL Model Interface initialized on {self.device}")
         
+<<<<<<< HEAD
     def predict(self, cob_features) -> Dict[str, Any]:
+=======
+    def to(self, device):
+        """PyTorch-style device movement method"""
+        self.device = device
+        self.model = self.model.to(device)
+        return self
+        
+    def predict(self, cob_features: np.ndarray) -> Dict[str, Any]:
+>>>>>>> d49a473ed6f4aef55bfdd47d6370e53582be6b7b
         """Make prediction using the model"""
         self.model.eval()
         with torch.no_grad():