remove pairwise policy

theislab · Oct 10, 2023 · 0419670 · 0419670
1 parent 501b018
commit 0419670
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/moscot/problems/generic/_generic.py b/src/moscot/problems/generic/_generic.py
@@ -479,7 +479,7 @@ def _base_problem_type(self) -> Type[B]:
 
     @property
     def _valid_policies(self) -> Tuple[Policy_t, ...]:
-        return _constants.SEQUENTIAL, _constants.PAIRWISE, _constants.EXPLICIT  # type: ignore[return-value]
+        return _constants.SEQUENTIAL, _constants.EXPLICIT  # type: ignore[return-value]
 
 
 class NeuralProblem(CompoundProblem[K, B], GenericAnalysisMixin[K, B]):
@@ -564,7 +564,7 @@ def _base_problem_type(self) -> Type["NeuralProblem[K, B]"]:  # type:ignore[over
 
     @property
     def _valid_policies(self) -> Tuple[Policy_t, ...]:
-        return _constants.SEQUENTIAL, _constants.PAIRWISE, _constants.EXPLICIT  # type: ignore[return-value]
+        return _constants.SEQUENTIAL, _constants.EXPLICIT  # type: ignore[return-value]
 
 
 class ConditionalNeuralProblem(CondOTProblem, GenericAnalysisMixin[K, B]):
@@ -649,7 +649,7 @@ def _base_problem_type(self) -> Type[CondOTProblem]:
 
     @property
     def _valid_policies(self) -> Tuple[Policy_t, ...]:
-        return _constants.SEQUENTIAL, _constants.PAIRWISE, _constants.EXPLICIT  # type: ignore[return-value]
+        return _constants.SEQUENTIAL, _constants.EXPLICIT  # type: ignore[return-value]
 
 
 class MGNeuralProblem(CompoundProblem[K, B], GenericAnalysisMixin[K, B]):
@@ -718,4 +718,4 @@ def _base_problem_type(self) -> Type["NeuralProblem[K, B]"]:  # type:ignore[over
 
     @property
     def _valid_policies(self) -> Tuple[Policy_t, ...]:
-        return _constants.SEQUENTIAL, _constants.PAIRWISE, _constants.EXPLICIT  # type: ignore[return-value]
+        return _constants.SEQUENTIAL, _constants.EXPLICIT  # type: ignore[return-value]