fix merge

This commit is contained in:
Dobromir Popov
2025-10-02 23:50:08 +03:00
parent 8654e08028
commit a468c75c47
13 changed files with 150 additions and 14309 deletions

View File

@@ -267,17 +267,6 @@ 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():

View File

@@ -4,11 +4,6 @@ import torch.optim as optim
import numpy as np
from collections import deque
import random
<<<<<<< HEAD
from typing import Tuple, List
=======
from typing import Tuple, List, Dict, Any
>>>>>>> d49a473ed6f4aef55bfdd47d6370e53582be6b7b
import os
import sys
import logging