remove useless kwargs

GFNOrg · Jan 24, 2025 · 752b937 · 752b937
1 parent 4312107
commit 752b937
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/gfn/gflownet/base.py b/src/gfn/gflownet/base.py
@@ -71,7 +71,7 @@ def to_training_samples(self, trajectories: Trajectories) -> TrainingSampleType:
         """Converts trajectories to training samples. The type depends on the GFlowNet."""
 
     @abstractmethod
-    def loss(self, env: Env, training_objects: Any, **kwargs: Any):
+    def loss(self, env: Env, training_objects: Any):
         """Computes the loss given the training objects."""
 
 
@@ -113,11 +113,7 @@ def pf_pb_parameters(self):
 
     @abstractmethod
     def loss(
-        self,
-        env: Env,
-        training_objects: Any,
-        recalculate_all_logprobs: bool = False,
-        **kwargs: Any,
+        self, env: Env, training_objects: Any, recalculate_all_logprobs: bool = False
     ):
         """Computes the loss given the training objects.