Merge branch 'main' of https://github.com/Nixtla/nixtla

Nixtla · Sep 23, 2023 · 6b104d4 · 6b104d4
2 parents fac7e28 + 1bcfa17
commit 6b104d4
Show file tree

Hide file tree

Showing 7 changed files with 1,748 additions and 329 deletions.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -48,11 +48,11 @@ jobs:
         uses: actions/checkout@v2
 
       - name: Set up environment
-        uses: mamba-org/provision-with-micromamba@main
+        uses: mamba-org/setup-micromamba@v1 
         with:
           environment-file: environment.yml
-          extra-specs: python=${{ matrix.python-version }}
-          cache-env: true
+          create-args: python=${{ matrix.python-version }}
+          cache-environment: true
 
       - name: Install pip requirements
         run: pip install ./ 
@@ -78,11 +78,11 @@ jobs:
         uses: actions/checkout@v2
 
       - name: Set up environment
-        uses: mamba-org/provision-with-micromamba@main
+        uses: mamba-org/setup-micromamba@v1 
         with:
           environment-file: environment.yml
-          extra-specs: python=${{ matrix.python-version }}
-          cache-env: true
+          create-args: python=${{ matrix.python-version }}
+          cache-environment: true
 
       - name: Install pip requirements
         run: pip install ./ 

diff --git a/nbs/data/pltr.csv b/nbs/data/pltr.csv
diff --git a/nbs/distributed.timegpt.ipynb b/nbs/distributed.timegpt.ipynb
@@ -196,7 +196,10 @@
     "            time_col: str = 'ds',\n",
     "            target_col: str = 'y',\n",
     "            level: Union[int, float] = 99,\n",
+    "            clean_ex_first: bool = True,\n",
     "            validate_token: bool = False,\n",
+    "            date_features: Union[bool, List[str]] = False,\n",
+    "            date_features_to_one_hot: Union[bool, List[str]] = True,\n",
     "            num_partitions: Optional[int] = None,\n",
     "        ) -> fugue.AnyDataFrame:\n",
     "        kwargs = dict(\n",
@@ -205,7 +208,10 @@
     "            time_col=time_col,\n",
     "            target_col=target_col,\n",
     "            level=level,\n",
+    "            clean_ex_first=clean_ex_first,\n",
     "            validate_token=validate_token,\n",
+    "            date_features=date_features,\n",
+    "            date_features_to_one_hot=date_features_to_one_hot,\n",
     "        )\n",
     "        schema = self._get_anomalies_schema(id_col=id_col, time_col=time_col)\n",
     "        anomalies_df = self._distribute_method(\n",
@@ -603,7 +609,8 @@
     "def test_anomalies_dataframe_diff_cols(df: fugue.AnyDataFrame, id_col: str = 'id_col', time_col: str = 'time_col'):\n",
     "    test_anomalies(df, id_col=id_col, time_col=time_col, num_partitions=1)\n",
     "    test_anomalies(df, id_col=id_col, time_col=time_col, level=90, num_partitions=1)\n",
-    "    test_anomalies_same_results_num_partitions(df, id_col=id_col, time_col=time_col)"
+    "    test_anomalies_same_results_num_partitions(df, id_col=id_col, time_col=time_col)\n",
+    "    test_anomalies_same_results_num_partitions(df, id_col=id_col, time_col=time_col, date_features=True, clean_ex_first=False)"
    ]
   },
   {