black and ruff

Author: Swandog

pylace/lace/__init__.py

@@ -1,6 +1,7 @@
 """The Python bindings for the Lace ML tool."""
 
 from importlib import metadata
+
 from lace import core
 from lace.codebook import Codebook
 from lace.core import (
@@ -37,4 +38,4 @@
     "ValueMap",
 ]
 
-__version__ = metadata.version('pylace')
+__version__ = metadata.version("pylace")

pylace/lace/analysis.py

@@ -307,6 +307,7 @@ def held_out_neglogp(
     │ ["Apogee_km", "Country_of_Contra… ┆ 2.992173            ┆ 7         │
     │ ["Apogee_km", "Class_of_Orbit", … ┆ 3.956759            ┆ 8         │
     └───────────────────────────────────┴─────────────────────┴───────────┘
+
     """
     search = (
         HoldOutSearchMethod.Greedy if greedy else HoldOutSearchMethod.Enumerate
@@ -420,6 +421,7 @@ def held_out_inconsistency(
     │ ["Apogee_km", "Country_of_Contra… ┆ 0.756218                  ┆ 7         │
     │ ["Apogee_km", "Class_of_Orbit", … ┆ 1.0                       ┆ 8         │
     └───────────────────────────────────┴───────────────────────────┴───────────┘
+
     """
     search = (
         HoldOutSearchMethod.Greedy if greedy else HoldOutSearchMethod.Enumerate
@@ -530,6 +532,7 @@ def held_out_uncertainty(
     │ ["Apogee_km", "Country_of_Contra… ┆ 0.077359                ┆ 7         │
     │ ["Apogee_km", "Class_of_Orbit", … ┆ 0.089026                ┆ 8         │
     └───────────────────────────────────┴─────────────────────────┴───────────┘
+
     """
     search = (
         HoldOutSearchMethod.Greedy if greedy else HoldOutSearchMethod.Enumerate
@@ -654,6 +657,7 @@ def attributable_inconsistency(
     ... )  # doctest: +NORMALIZE_WHITESPACE
     >>> frac
     0.2930260843667006
+
     """
 
     return _attributable_holdout(
@@ -727,6 +731,7 @@ def attributable_neglogp(
     ... )  # doctest: +NORMALIZE_WHITESPACE
     >>> frac
     0.29302608436670047
+
     """
 
     return _attributable_holdout(
@@ -797,6 +802,7 @@ def attributable_uncertainty(
     ... )  # doctest: +NORMALIZE_WHITESPACE
     >>> frac
     0.1814171785207335
+
     """
 
     return _attributable_holdout(
@@ -981,6 +987,7 @@ def explain_prediction(
     │ Source_Used_for_Orbital_Data ┆ 0.001454  │
     │ Inclination_radians          ┆ 0.057333  │
     └──────────────────────────────┴───────────┘
+
     """
     if method is None:
         method = ABLATIVE_ERR

pylace/lace/codebook.py

@@ -50,6 +50,7 @@ class Codebook:
     ----------
     codebook: core.Codebook
         A reference to the underlying rust data structure
+
     """
 
     codebook: _lc.Codebook
@@ -89,6 +90,7 @@ def from_df(
         no_hypers: bool, optional
             If `True`, disable hyper prior inference. Priors will be derived
             from the data and will remain static.
+
         """
 
         if isinstance(df, pd.DataFrame):
@@ -214,6 +216,7 @@ def column_metadata(self) -> _ColumnMetadataIndexer:
         }
         >>> codebook.column_names[-5:]
         ['solitary', 'nestspot', 'domestic', 'number-in-wild', 'scary']
+
         """
         return _ColumnMetadataIndexer(self.codebook)
 
@@ -228,6 +231,7 @@ def shape(self) -> Tuple[int, int]:
         >>> codebook = Animals().codebook
         >>> codebook.shape
         (50, 85)
+
         """
         return self.codebook.shape
 
@@ -242,6 +246,7 @@ def row_names(self) -> List[str]:
         >>> codebook = Animals().codebook
         >>> codebook.row_names[:4]
         ['antelope', 'grizzly+bear', 'killer+whale', 'beaver']
+
         """
         return self.codebook.row_names
 
@@ -256,6 +261,7 @@ def column_names(self) -> List[str]:
         >>> codebook = Animals().codebook
         >>> codebook.column_names[:4]
         ['black', 'white', 'blue', 'brown']
+
         """
         return self.codebook.column_names
 
@@ -284,6 +290,7 @@ def rename(self, name: str):
           view_alpha_prior: G(α: 1, β: 1)
           columns: 85
           rows: 50
+
         """
         codebook = copy.copy(self)
         codebook.codebook.rename(name)
@@ -318,6 +325,7 @@ def set_state_alpha_prior(self, shape: float = 1.0, rate: float = 1.0):
           view_alpha_prior: G(α: 1, β: 1)
           columns: 85
           rows: 50
+
         """
         codebook = copy.copy(self)
         codebook.codebook.set_state_alpha_prior(shape, rate)
@@ -352,6 +360,7 @@ def set_view_alpha_prior(self, shape: float = 1.0, rate: float = 1.0):
           view_alpha_prior: G(α: 2, β: 3.1)
           columns: 85
           rows: 50
+
         """
         codebook = copy.copy(self)
         codebook.codebook.set_view_alpha_prior(shape, rate)
@@ -392,6 +401,7 @@ def set_row_names(
         >>> new_rows = np.array(["one", "two", "three-hahaha"])
         >>> codebook.set_row_names(new_rows).row_names
         ['one', 'two', 'three-hahaha']
+
         """
         if isinstance(row_names, (pl.Series, pd.Series)):
             row_names = row_names.to_list()
@@ -433,6 +443,7 @@ def value_map(self, col: str):
         ValueMap (UInt, k=2)
         >>> [c for c in vm.values()]
         [0, 1]
+
         """
         return self.column_metadata[col].value_map