debugged DNN.init() to correctly accept 4 arguments when it was previ…

…ously missing a 'max_control_cost' argument (#89)
DynaDojo · Jul 10, 2024 · dd9efe0 · dd9efe0
1 parent 408e564
commit dd9efe0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/dynadojo/baselines/dnn.py b/src/dynadojo/baselines/dnn.py
@@ -170,8 +170,9 @@ class DNN(TorchBaseClass):
     def __init__(self, 
             embed_dim,
             timesteps,
+            max_control_cost,
             **kwargs):
-        super().__init__(embed_dim, timesteps, **kwargs)
+        super().__init__(embed_dim, timesteps, max_control_cost, **kwargs)
         self.model = torch.nn.Sequential(
             torch.nn.Linear(self.embed_dim, embed_dim*10),
             torch.nn.ReLU(),