LPX55
commited on
Commit
·
9310be4
1
Parent(s):
401d88f
fix
Browse files
agents/ensemble_weights.py
CHANGED
|
@@ -3,7 +3,7 @@ import torch
|
|
| 3 |
from utils.registry import MODEL_REGISTRY # Import MODEL_REGISTRY
|
| 4 |
from utils.agent_logger import AgentLogger
|
| 5 |
|
| 6 |
-
|
| 7 |
logger = logging.getLogger(__name__)
|
| 8 |
|
| 9 |
class ContextualWeightOverrideAgent:
|
|
|
|
| 3 |
from utils.registry import MODEL_REGISTRY # Import MODEL_REGISTRY
|
| 4 |
from utils.agent_logger import AgentLogger
|
| 5 |
|
| 6 |
+
agent_logger = AgentLogger()
|
| 7 |
logger = logging.getLogger(__name__)
|
| 8 |
|
| 9 |
class ContextualWeightOverrideAgent:
|