remove slice sampler, fit returns self

pymc-labs · Oct 20, 2023 · c8e9fa1 · c8e9fa1
1 parent 03fdcb3
commit c8e9fa1
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/pymc_marketing/clv/models/pareto_nbd.py b/pymc_marketing/clv/models/pareto_nbd.py
@@ -1,5 +1,5 @@
 import warnings
-from typing import Any, Dict, Optional, Sequence, Tuple, TypeVar, Union
+from typing import Any, Dict, Optional, Sequence, Tuple, Union
 
 import numpy as np
 import pandas as pd
@@ -20,8 +20,6 @@
 from pymc_marketing.clv.models.basic import CLVModel
 from pymc_marketing.clv.utils import to_xarray
 
-SELF = TypeVar("SELF")
-
 
 @node_rewriter([Elemwise])
 def local_reduce_max_num_iters_hyp2f1_grad(fgraph, node):
@@ -326,7 +324,7 @@ def fit(self, fit_method: str = "map", **kwargs):  # type: ignore
                     action="ignore",
                     category=UserWarning,
                 )
-                super().fit(fit_method, **kwargs)  # type: ignore
+                super().fit(fit_method, **kwargs)
 
         return self