fix: new black version

omg this is horrible
Nixtla · Mar 22, 2024 · 0568e04 · 0568e04
1 parent 0aa6eb7
commit 0568e04
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/nixtlats/distributed/timegpt.py b/nixtlats/distributed/timegpt.py
@@ -47,6 +47,7 @@ def _cotransform(
 
 # %% ../../nbs/distributed.timegpt.ipynb 4
 class _DistributedTimeGPT:
+
     def __init__(
         self,
         token: Optional[str] = None,

diff --git a/nixtlats/timegpt.py b/nixtlats/timegpt.py
@@ -97,6 +97,7 @@
 
 # %% ../nbs/timegpt.ipynb 6
 class _TimeGPTModel:
+
     def __init__(
         self,
         client: Nixtla,
@@ -581,9 +582,11 @@ def detect_anomalies(self, df: pd.DataFrame):
                 y=y,
                 x=x,
                 freq=self.freq,
-                level=[self.level]
-                if (isinstance(self.level, int) or isinstance(self.level, float))
-                else [self.level[0]],
+                level=(
+                    [self.level]
+                    if (isinstance(self.level, int) or isinstance(self.level, float))
+                    else [self.level[0]]
+                ),
                 clean_ex_first=self.clean_ex_first,
                 model=self.model,
             ),
@@ -1030,6 +1033,7 @@ def plot(
 
 # %% ../nbs/timegpt.ipynb 11
 class TimeGPT(_TimeGPT):
+
     def _instantiate_distributed_timegpt(self):
         from nixtlats.distributed.timegpt import _DistributedTimeGPT