fix colon bug

landoskape · Apr 17, 2024 · 4e9ed50 · 4e9ed50
1 parent 8533150
commit 4e9ed50
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dominoes/datasets/base.py b/dominoes/datasets/base.py
@@ -332,7 +332,7 @@ def reward_function(self, choices, batch, **kwargs):
         if self.task == "sequencer":
             return self._measurereward_sequencer(choices, batch, **kwargs)
         elif self.task == "sorting":
-            return self._measurereward_sorter(choices, batch, **kwargs):
+            return self._measurereward_sorter(choices, batch, **kwargs)
         else:
             raise ValueError(f"task {self.task} not recognized")