Skip improt

skyro/_mixin.py

@@ -1,4 +1,3 @@
-import sys
 from functools import cached_property
 from operator import attrgetter
 from random import randint
@@ -10,11 +9,6 @@
 from numpyro.infer import MCMC, NUTS
 from numpyro.infer.mcmc import MCMCKernel
 
-if sys.version_info >= (3, 11):
-    pass
-else:
-    pass
-
 from ._result import NumpyroResultSet
 from .exc import ConvergenceError
 

skyro/sklearn.py

@@ -1,16 +1,10 @@
-import sys
 from typing import Any, Dict
 
 import numpy as np
 from numpyro.infer import Predictive
 from skbase.base import BaseEstimator
 
-if sys.version_info >= (3, 11):
-    from typing import Self
-else:
-    from typing_extensions import Self
-
-from skyro._mixin import BaseNumpyroMixin
+from ._mixin import BaseNumpyroMixin
 
 
 class BaseNumpyroEstimator(BaseNumpyroMixin, BaseEstimator):
@@ -88,5 +82,5 @@ def predict(self, X, full_posterior: bool = False, **kwargs):
 
         return self.reduce(output)
 
-    def sample_prior_predictive(self, X, **kwargs) -> Self:
+    def sample_prior_predictive(self, X, **kwargs):
         return self._do_sample(X=X)