From 4e9ed5048c6e5d32a14fbec9c8d275b9f3d0b934 Mon Sep 17 00:00:00 2001 From: landoskape Date: Wed, 17 Apr 2024 08:38:18 +0100 Subject: [PATCH] fix colon bug --- dominoes/datasets/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dominoes/datasets/base.py b/dominoes/datasets/base.py index 73e8265..4a4c8f9 100644 --- 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")