Skip to content

Commit

Permalink
Make pylint happy
Browse files Browse the repository at this point in the history
sorry for doing this on main
  • Loading branch information
vankesteren committed Feb 23, 2024
1 parent 1b664eb commit a2e5e71
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions metasyncontrib/disclosure/constant.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
"Module for constant distributions"
import polars as pl
from metasyncontrib.disclosure.base import metadist_disclosure
from metasyn.distribution.constant import (
ConstantDistribution,
DiscreteConstantDistribution,
Expand All @@ -9,6 +8,7 @@
TimeConstantDistribution,
DateConstantDistribution,
)
from metasyncontrib.disclosure.base import metadist_disclosure


def disclosure_constant(cls):
Expand All @@ -25,9 +25,9 @@ def _fit(values: pl.Series, n_avg=11):

if count >= n_avg:
return cls(value)

return cls.default_distribution()

setattr(cls, "_fit", _fit)
return cls

Expand Down Expand Up @@ -61,4 +61,3 @@ class DisclosureTimeConstant(TimeConstantDistribution):
@disclosure_constant
class DisclosureDateConstant(DateConstantDistribution):
"Disclosure controlled DateConstantDistribution"

0 comments on commit a2e5e71

Please sign in to comment.