add pre-commit

.github/workflows/lint.yaml

@@ -0,0 +1,25 @@
+name: Lint
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Clone repo
+        uses: actions/checkout@v2
+
+      - name: Set up python
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.10'
+
+      - name: Install dependencies
+        run: pip install black nbdev pre-commit
+
+      - name: Run pre-commit
+        run: pre-commit run

CONTRIBUTING.md

@@ -59,6 +59,7 @@ Please write to `[email protected]` if you're insterested in contributing to this pr
 Before doing any changes to the code, please install the git hooks that run automatic scripts during each commit and merge to strip the notebooks of superfluous metadata (and avoid merge conflicts).
 ```
 nbdev_install_hooks
+pre-commit install
 ```
 
 ### Preview Changes

nbs/docs/tutorials/5_finetuning.ipynb

@@ -307,10 +307,6 @@
    "display_name": "python3",
    "language": "python",
    "name": "python3"
-  },
-  "language_info": {
-   "name": "python",
-   "version": "3.10.13"
   }
  },
  "nbformat": 4,

nixtlats/distributed/timegpt.py

@@ -47,6 +47,7 @@ def _cotransform(
 
 # %% ../../nbs/distributed.timegpt.ipynb 4
 class _DistributedTimeGPT:
+
     def __init__(
         self,
         token: Optional[str] = None,

nixtlats/timegpt.py

@@ -90,6 +90,7 @@
 
 # %% ../nbs/timegpt.ipynb 6
 class _TimeGPTModel:
+
     def __init__(
         self,
         client: Nixtla,
@@ -575,9 +576,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,
             ),
@@ -1024,6 +1027,7 @@ def plot(
 
 # %% ../nbs/timegpt.ipynb 11
 class TimeGPT(_TimeGPT):
+
     def _instantiate_distributed_timegpt(self):
         from nixtlats.distributed.timegpt import _DistributedTimeGPT
 

setup.py

@@ -3,7 +3,7 @@
 with open("README.md", "r", encoding="utf8") as fh:
     long_description = fh.read()
 
-dev = ["black", "nbdev", "plotly", "python-dotenv", "statsforecast", "datasetsforecast"]
+dev = ["black", "datasetsforecast", "nbdev", "plotly", "python-dotenv", "statsforecast", "pre-commit"]
 distributed = ["dask[dataframe]", "fugue[ray]>=0.8.7", "pyspark", "ray[serve-grpc]"]
 plotting = ["utilsforecast[plotting]>=0.0.5"]
 date_extras = ["holidays"]