diff --git a/.fernignore b/.fernignore
index 55d6d8cfa..d8d6ea52a 100644
--- a/.fernignore
+++ b/.fernignore
@@ -18,9 +18,9 @@ environment.yml
settings.ini
setup.py
pyproject.toml
-nixtlats/nixtla_client.py
-nixtlats/utils.py
-nixtlats/_modidx.py
-nixtlats/__init__.py
-nixtlats/distributed/
-nixtlats/date_features.py
+nixtla/nixtla_client.py
+nixtla/utils.py
+nixtla/_modidx.py
+nixtla/__init__.py
+nixtla/distributed/
+nixtla/date_features.py
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index ec97b7306..2a4468779 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -57,8 +57,8 @@ jobs:
- name: Check import
run: |
- python -c "from nixtlats import TimeGPT;"
- python -c "from nixtlats import NixtlaClient;"
+ python -c "from nixtla import TimeGPT;"
+ python -c "from nixtla import NixtlaClient;"
run-tests:
runs-on: ${{ matrix.os }}
diff --git a/.github/workflows/models-performance.yaml b/.github/workflows/models-performance.yaml
index e787949e3..cb006cb93 100644
--- a/.github/workflows/models-performance.yaml
+++ b/.github/workflows/models-performance.yaml
@@ -34,7 +34,7 @@ jobs:
cache-environment: true
- name: Install pip requirements
- run: pip install ./
+ run: pip install '.[plotting]'
- name: Run evaluation
run: python -m action_files.models_performance.main
diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index ec703542b..acffb0c0e 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -1,39 +1,39 @@
-# This workflow will upload a Python Package using Twine when a release is created
-# For more information see: https://help.github.com/en/actions/language-and-framework-guides/using-python-with-github-actions#publishing-to-package-registries
-
-# This workflow uses actions that are not certified by GitHub.
-# They are provided by a third-party and are governed by
-# separate terms of service, privacy policy, and support
-# documentation.
-
-name: Upload Python Package
+name: Upload Python Packages
on:
- release:
- types: [published]
-
-permissions:
- contents: read
+ push:
+ tags: ['v*']
jobs:
deploy:
-
runs-on: ubuntu-latest
-
+ permissions:
+ id-token: write
steps:
- uses: actions/checkout@v3
+
- name: Set up Python
- uses: actions/setup-python@v3
+ uses: actions/setup-python@v4
with:
- python-version: '3.x'
+ python-version: '3.10'
+
- name: Install dependencies
- run: |
- python -m pip install --upgrade pip
- pip install build
- - name: Build package
+ run: python -m pip install --upgrade pip && pip install build
+
+ - name: Build nixtla package
run: python -m build
- - name: Publish package
- uses: pypa/gh-action-pypi-publish@27b31702a0e7fc50959f5ad993c78deac1bdfc29
- with:
- user: __token__
- password: ${{ secrets.PYPI_API_TOKEN }}
+
+ - name: Publish nixtla package
+ uses: pypa/gh-action-pypi-publish@release/v1
+
+ - name: Build nixtlats package
+ run: >
+ rm -rf build dist &&
+ mv nixtla nixtlats &&
+ sed -i 's/name="nixtla"/name="nixtlats"/g' setup.py &&
+ find nixtlats -type f -name '*.py' -exec sed -i 's/from nixtla/from nixtlats/g' {} + &&
+ echo -e 'import warnings\nwarnings.warn("This package is deprecated, please install nixtla instead.", category=FutureWarning)' >> nixtlats/__init__.py &&
+ python -m build
+
+ - name: Publish nixtlats package
+ uses: pypa/gh-action-pypi-publish@release/v1
diff --git a/.github/workflows/test-publish.yml b/.github/workflows/test-publish.yml
new file mode 100644
index 000000000..af6cb77b9
--- /dev/null
+++ b/.github/workflows/test-publish.yml
@@ -0,0 +1,42 @@
+name: Upload Python Packages to TestPyPI
+
+on:
+ workflow_dispatch:
+
+jobs:
+ deploy:
+ runs-on: ubuntu-latest
+ permissions:
+ id-token: write
+ steps:
+ - uses: actions/checkout@v3
+
+ - name: Set up Python
+ uses: actions/setup-python@v4
+ with:
+ python-version: '3.10'
+
+ - name: Install dependencies
+ run: python -m pip install --upgrade pip && pip install build
+
+ - name: Build nixtla package
+ run: python -m build
+
+ - name: Publish nixtla package
+ uses: pypa/gh-action-pypi-publish@release/v1
+ with:
+ repository-url: https://test.pypi.org/legacy/
+
+ - name: Build nixtlats package
+ run: >
+ rm -rf build dist &&
+ mv nixtla nixtlats &&
+ sed -i 's/name="nixtla"/name="nixtlats"/g' setup.py &&
+ find nixtlats -type f -name '*.py' -exec sed -i 's/from nixtla/from nixtlats/g' {} + &&
+ echo -e 'import warnings\nwarnings.warn("This package is deprecated, please install nixtla instead.", category=FutureWarning)' >> nixtlats/__init__.py &&
+ python -m build
+
+ - name: Publish nixtlats package
+ uses: pypa/gh-action-pypi-publish@release/v1
+ with:
+ repository-url: https://test.pypi.org/legacy/
diff --git a/.gitignore b/.gitignore
index 54af80fe1..fee63154a 100644
--- a/.gitignore
+++ b/.gitignore
@@ -25,3 +25,4 @@ nbs/.last_checked
.env
*/summary.md
*/*.png
+longhorizon
diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md
index c2235c593..d6e80b950 100644
--- a/CONTRIBUTING.md
+++ b/CONTRIBUTING.md
@@ -38,13 +38,13 @@ The repo comes with an `environment.yml` file which contains the libraries neede
Once you have `conda` go to the top level directory of the repository and run the following lines (we recommend to use `python=3.10` for development):
```
-conda create -n nixtlats python=3.10
-conda activate nixtlats
+conda create -n nixtla python=3.10
+conda activate nixtla
conda env update -f environment.yml
```
#### Install the library
-Once you have your environment setup, activate it using `conda activate nixtlats` and then install the library in editable mode using `pip install -e ".[dev]"`
+Once you have your environment setup, activate it using `conda activate nixtla` and then install the library in editable mode using `pip install -e ".[dev]"`
#### Set Up TimeGPT token
This library uses `python-dotenv` for development. To set up your TimeGPT token, add the following lines to your `.env` file:
diff --git a/README.md b/README.md
index 9c61a36fe..1603cd401 100644
--- a/README.md
+++ b/README.md
@@ -2,17 +2,17 @@

-
NixtlaTS
+
Nixtla
Forecast using TimeGPT
[](https://github.com/Nixtla/nixtla/actions/workflows/ci.yaml)
-[](https://pypi.org/project/nixtlats/)
-[](https://pypi.org/project/nixtlats/)
-[](https://github.com/Nixtla/nixtlats/blob/main/LICENSE)
+[](https://pypi.org/project/nixtla/)
+[](https://pypi.org/project/nixtla/)
+[](https://github.com/Nixtla/nixtla/blob/main/LICENSE)
[](https://nixtla.github.io/nixtla/)
-[](https://pepy.tech/project/nixtlats)
+[](https://pepy.tech/project/nixtla)
-**NixtlaTS** offers a collection of classes and methods to interact with the API of TimeGPT.
+**Nixtla** offers a collection of classes and methods to interact with the API of TimeGPT.
# π°οΈ TimeGPT: Revolutionizing Time-Series Analysis
@@ -25,16 +25,16 @@ In seconds, TimeGPT can discern complex patterns and predict future data points,
In addition to its core capabilities, TimeGPT supports fine-tuning, enhancing its specialization for specific prediction tasks. π― This feature is like training a machine learning model on a targeted data subset to improve its task-specific performance, making TimeGPT an even more versatile tool for your predictive needs.
-## π `NixtlaTS`: Your Gateway to TimeGPT
+## π `Nixtla`: Your Gateway to TimeGPT
-With `NixtlaTS`, you can easily interact with TimeGPT through simple API calls, making the power of TimeGPT readily accessible in your projects.
+With `Nixtla`, you can easily interact with TimeGPT through simple API calls, making the power of TimeGPT readily accessible in your projects.
## π» Installation
-Get `NixtlaTS` up and running with a simple pip command:
+Get `Nixtla` up and running with a simple pip command:
```python
-pip install nixtlats>=0.1.0
+pip install nixtla>=0.1.0
```
## π Quick Start
@@ -44,7 +44,7 @@ Get started with TimeGPT now:
```python
df = pd.read_csv('https://raw.githubusercontent.com/Nixtla/transfer-learning-time-series/main/datasets/electricity-short.csv')
-from nixtlats import NixtlaClient
+from nixtla import NixtlaClient
nixtla = NixtlaClient(
# defaults to os.environ.get("NIXTLA_API_KEY")
api_key = 'my_api_key_provided_by_nixtla'
diff --git a/action_files/models_performance/main.py b/action_files/models_performance/main.py
index b0a7e1967..e4353719f 100644
--- a/action_files/models_performance/main.py
+++ b/action_files/models_performance/main.py
@@ -11,7 +11,7 @@
from utilsforecast.evaluation import evaluate
from utilsforecast.losses import mae, mape, mse
-from nixtlats import NixtlaClient
+from nixtla import NixtlaClient
logger = logging.getLogger(__name__)
@@ -144,9 +144,11 @@ def evaluate_timegpt(self, model: str) -> Tuple[pd.DataFrame, pd.DataFrame]:
timegpt = NixtlaClient()
fcst_df = timegpt.forecast(
df=self.df_train,
- X_df=self.df_test.drop(columns=self.target_col)
- if self.df.shape[1] > 3
- else None,
+ X_df=(
+ self.df_test.drop(columns=self.target_col)
+ if self.df.shape[1] > 3
+ else None
+ ),
h=self.h,
freq=self.freq,
level=self.level,
diff --git a/environment.yml b/environment.yml
index b0af3fdb6..103674d97 100644
--- a/environment.yml
+++ b/environment.yml
@@ -1,12 +1,11 @@
-name: nixtlats
+name: nixtla
channels:
- conda-forge
dependencies:
- - dask
+ - dask<=2024.2.1
- jupyterlab
- pandas<2.2
- plotly
- - prophet
- pyspark>=3.3
- pip:
- black
diff --git a/experiments/azure-automl-forecasting/requirements.txt b/experiments/azure-automl-forecasting/requirements.txt
index 56d9b64e6..df9515fed 100644
--- a/experiments/azure-automl-forecasting/requirements.txt
+++ b/experiments/azure-automl-forecasting/requirements.txt
@@ -3,7 +3,7 @@ azure-identity
azureml-core
fire
mltable
-nixtlats
+nixtla
pandas
python-dotenv
rich
diff --git a/experiments/azure-automl-forecasting/src/nixtla_timegpt.py b/experiments/azure-automl-forecasting/src/nixtla_timegpt.py
index 16571613a..1edbc5c15 100644
--- a/experiments/azure-automl-forecasting/src/nixtla_timegpt.py
+++ b/experiments/azure-automl-forecasting/src/nixtla_timegpt.py
@@ -4,7 +4,7 @@
import fire
from dotenv import load_dotenv
-from nixtlats import TimeGPT
+from nixtla import TimeGPT
from src.utils.data_handler import ExperimentDataset, ForecastDataset
diff --git a/experiments/prophet/environment.yml b/experiments/prophet/environment.yml
index b4f86021e..69cf1e3ec 100644
--- a/experiments/prophet/environment.yml
+++ b/experiments/prophet/environment.yml
@@ -8,7 +8,7 @@ dependencies:
- python=3.10
- pip:
- fire
- - nixtlats
+ - nixtla
- python-dotenv
- statsforecast
- utilsforecast
diff --git a/experiments/prophet/src/timegpt_exp.py b/experiments/prophet/src/timegpt_exp.py
index d363d0b97..a7ee7d945 100644
--- a/experiments/prophet/src/timegpt_exp.py
+++ b/experiments/prophet/src/timegpt_exp.py
@@ -3,7 +3,7 @@
import fire
from dotenv import load_dotenv
-from nixtlats import TimeGPT
+from nixtla import TimeGPT
from src.tools import ExperimentHandler
diff --git a/nbs/_quarto.yml b/nbs/_quarto.yml
index 7b306ecde..303d82636 100644
--- a/nbs/_quarto.yml
+++ b/nbs/_quarto.yml
@@ -46,7 +46,7 @@ website:
href: https://join.slack.com/t/nixtlaworkspace/shared_invite/zt-135dssye9-fWTzMpv2WBthq8NK0Yvu6A
right:
- icon: github
- href: "https://github.com/nixtla/nixtlats"
+ href: "https://github.com/nixtla/nixtla"
- icon: twitter
href: https://twitter.com/nixtlainc
aria-label: Nixtla Twitter
@@ -54,6 +54,6 @@ website:
sidebar:
style: floating
body-footer: |
- Give us a βΒ on [Github](https://github.com/nixtla/nixtlats)
+ Give us a βΒ on [Github](https://github.com/nixtla/nixtla)
metadata-files: [nbdev.yml, sidebar.yml]
diff --git a/nbs/date_features.ipynb b/nbs/date_features.ipynb
index 11ba04b2a..2b07251d5 100644
--- a/nbs/date_features.ipynb
+++ b/nbs/date_features.ipynb
@@ -120,7 +120,7 @@
" except ModuleNotFoundError:\n",
" raise Exception(\n",
" 'You have to install additional libraries to use holidays, '\n",
- " 'please install them using `pip install \"nixtlats[date_extras]\"`'\n",
+ " 'please install them using `pip install \"nixtla[date_extras]\"`'\n",
" )\n",
" return _get_holidays_df(dates, self.countries, country_holidays, list_supported_countries())\n",
" \n",
diff --git a/nbs/distributed.nixtla_client.ipynb b/nbs/distributed.nixtla_client.ipynb
index db3c51132..2f4afa734 100644
--- a/nbs/distributed.nixtla_client.ipynb
+++ b/nbs/distributed.nixtla_client.ipynb
@@ -301,7 +301,7 @@
" return fcst_df\n",
" \n",
" def _instantiate_nixtla_client(self):\n",
- " from nixtlats.nixtla_client import _NixtlaClient\n",
+ " from nixtla.nixtla_client import _NixtlaClient\n",
" nixtla_client = _NixtlaClient(\n",
" api_key=self.api_key, \n",
" base_url=self.base_url,\n",
diff --git a/nbs/docs/getting-started/1_getting_started_short.ipynb b/nbs/docs/getting-started/1_getting_started_short.ipynb
index 269123165..a961714cd 100644
--- a/nbs/docs/getting-started/1_getting_started_short.ipynb
+++ b/nbs/docs/getting-started/1_getting_started_short.ipynb
@@ -21,7 +21,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -94,7 +94,7 @@
"metadata": {},
"outputs": [],
"source": [
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/docs/getting-started/2_setting_up_your_authentication_api_key.ipynb b/nbs/docs/getting-started/2_setting_up_your_authentication_api_key.ipynb
index 900694d70..88f7047fd 100644
--- a/nbs/docs/getting-started/2_setting_up_your_authentication_api_key.ipynb
+++ b/nbs/docs/getting-started/2_setting_up_your_authentication_api_key.ipynb
@@ -32,7 +32,7 @@
"metadata": {},
"outputs": [],
"source": [
- "from nixtlats import NixtlaClient \n",
+ "from nixtla import NixtlaClient \n",
"nixtla_client = NixtlaClient(api_key ='your API key here')"
]
},
@@ -75,7 +75,7 @@
"metadata": {},
"outputs": [],
"source": [
- "from nixtlats import NixtlaClient\n",
+ "from nixtla import NixtlaClient\n",
"nixtla_client = NixtlaClient()"
]
},
@@ -137,7 +137,7 @@
"from dotenv import load_dotenv\n",
"load_dotenv()\n",
"\n",
- "from nixtlats import NixtlaClient\n",
+ "from nixtla import NixtlaClient\n",
"nixtla_client = NixtlaClient()"
]
},
diff --git a/nbs/docs/getting-started/3_azure_ai.ipynb b/nbs/docs/getting-started/3_azure_ai.ipynb
index 931a7869c..c5ea84b84 100644
--- a/nbs/docs/getting-started/3_azure_ai.ipynb
+++ b/nbs/docs/getting-started/3_azure_ai.ipynb
@@ -6,7 +6,7 @@
"source": [
"# AzureAI (coming soon)\n",
"\n",
- "> The foundational models for time series by Nixtla can be deployed on your Azure subscription. This page explains how to easily get started with TimeGEN deployed as an Azure AI endpoint. If you use the `nixtlats` library, it should be a drop-in replacement where you only need to change the client parameters (endpoint URL, API key, model name)."
+ "> The foundational models for time series by Nixtla can be deployed on your Azure subscription. This page explains how to easily get started with TimeGEN deployed as an Azure AI endpoint. If you use the `nixtla` library, it should be a drop-in replacement where you only need to change the client parameters (endpoint URL, API key, model name)."
]
},
{
@@ -50,7 +50,7 @@
"metadata": {},
"source": [
"```bash\n",
- "pip install nixtlats\n",
+ "pip install nixtla\n",
"```"
]
},
@@ -60,7 +60,7 @@
"source": [
"```python\n",
"import os\n",
- "from nixtlats import NixtlaClient\n",
+ "from nixtla import NixtlaClient\n",
"\n",
"base_url = os.environ[\"AZURE_AI_NIXTLA_BASE_URL\"]\n",
"api_key = os.environ[\"AZURE_AI_NIXTLA_API_KEY\"]\n",
diff --git a/nbs/docs/how-to-guides/0_distributed_fcst_spark.ipynb b/nbs/docs/how-to-guides/0_distributed_fcst_spark.ipynb
index 8b25be10d..054d0335d 100644
--- a/nbs/docs/how-to-guides/0_distributed_fcst_spark.ipynb
+++ b/nbs/docs/how-to-guides/0_distributed_fcst_spark.ipynb
@@ -19,7 +19,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -59,7 +59,7 @@
"id": "f2854f3c-7dc4-4615-9a85-7d7762fea647",
"metadata": {},
"source": [
- "As long as Spark is installed and configured, `TimeGPT` will be able to use it. If executing on a distributed Spark cluster, make use the `nixtlats` library is installed across all the workers."
+ "As long as Spark is installed and configured, `TimeGPT` will be able to use it. If executing on a distributed Spark cluster, make use the `nixtla` library is installed across all the workers."
]
},
{
@@ -120,7 +120,7 @@
"metadata": {},
"outputs": [],
"source": [
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/docs/how-to-guides/1_distributed_cv_spark.ipynb b/nbs/docs/how-to-guides/1_distributed_cv_spark.ipynb
index b75458336..dc64e4af8 100644
--- a/nbs/docs/how-to-guides/1_distributed_cv_spark.ipynb
+++ b/nbs/docs/how-to-guides/1_distributed_cv_spark.ipynb
@@ -19,7 +19,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -59,7 +59,7 @@
"id": "f2854f3c-7dc4-4615-9a85-7d7762fea647",
"metadata": {},
"source": [
- "As long as Spark is installed and configured, `TimeGPT` will be able to use it. If executing on a distributed Spark cluster, make use the `nixtlats` library is installed across all the workers."
+ "As long as Spark is installed and configured, `TimeGPT` will be able to use it. If executing on a distributed Spark cluster, make use the `nixtla` library is installed across all the workers."
]
},
{
@@ -120,7 +120,7 @@
"metadata": {},
"outputs": [],
"source": [
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/docs/tutorials/0_anomaly_detection.ipynb b/nbs/docs/tutorials/0_anomaly_detection.ipynb
index b9ee47666..5109d8b8c 100644
--- a/nbs/docs/tutorials/0_anomaly_detection.ipynb
+++ b/nbs/docs/tutorials/0_anomaly_detection.ipynb
@@ -24,7 +24,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -93,7 +93,7 @@
"outputs": [],
"source": [
"import pandas as pd\n",
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
@@ -800,7 +800,7 @@
"metadata": {},
"outputs": [],
"source": [
- "from nixtlats.date_features import CountryHolidays"
+ "from nixtla.date_features import CountryHolidays"
]
},
{
diff --git a/nbs/docs/tutorials/10_quantile_forecasts.ipynb b/nbs/docs/tutorials/10_quantile_forecasts.ipynb
index d0c6706f6..50b8adbbb 100644
--- a/nbs/docs/tutorials/10_quantile_forecasts.ipynb
+++ b/nbs/docs/tutorials/10_quantile_forecasts.ipynb
@@ -28,7 +28,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -99,7 +99,7 @@
"outputs": [],
"source": [
"import pandas as pd\n",
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/docs/tutorials/11_loss_function_finetuning.ipynb b/nbs/docs/tutorials/11_loss_function_finetuning.ipynb
index ac9434021..01745a4ec 100644
--- a/nbs/docs/tutorials/11_loss_function_finetuning.ipynb
+++ b/nbs/docs/tutorials/11_loss_function_finetuning.ipynb
@@ -29,7 +29,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -93,7 +93,7 @@
"outputs": [],
"source": [
"import pandas as pd\n",
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/docs/tutorials/12_longhorizon.ipynb b/nbs/docs/tutorials/12_longhorizon.ipynb
index fa528c63e..6d467d59f 100644
--- a/nbs/docs/tutorials/12_longhorizon.ipynb
+++ b/nbs/docs/tutorials/12_longhorizon.ipynb
@@ -27,7 +27,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -90,7 +90,7 @@
"metadata": {},
"outputs": [],
"source": [
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/docs/tutorials/1_exogenous_variables.ipynb b/nbs/docs/tutorials/1_exogenous_variables.ipynb
index b7757d65e..1b0cf96ad 100644
--- a/nbs/docs/tutorials/1_exogenous_variables.ipynb
+++ b/nbs/docs/tutorials/1_exogenous_variables.ipynb
@@ -28,7 +28,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -96,7 +96,7 @@
"outputs": [],
"source": [
"import pandas as pd\n",
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
@@ -644,7 +644,7 @@
"metadata": {},
"outputs": [],
"source": [
- "from nixtlats.date_features import CountryHolidays"
+ "from nixtla.date_features import CountryHolidays"
]
},
{
diff --git a/nbs/docs/tutorials/2_holidays.ipynb b/nbs/docs/tutorials/2_holidays.ipynb
index e16866cd8..151072095 100644
--- a/nbs/docs/tutorials/2_holidays.ipynb
+++ b/nbs/docs/tutorials/2_holidays.ipynb
@@ -26,7 +26,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -96,7 +96,7 @@
"source": [
"import pandas as pd\n",
"import numpy as np\n",
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
@@ -249,6 +249,7 @@
{
"cell_type": "code",
"execution_count": null,
+ "id": "f0af006e",
"metadata": {},
"outputs": [
{
@@ -402,7 +403,7 @@
}
],
"source": [
- "from nixtlats.date_features import CountryHolidays\n",
+ "from nixtla.date_features import CountryHolidays\n",
"\n",
"us_holidays = CountryHolidays(countries=['US'])\n",
"dates = pd.date_range(start=future_df.iloc[0]['month'], end=future_df.iloc[-1]['month'], freq='D')\n",
diff --git a/nbs/docs/tutorials/4_prediction_intervals.ipynb b/nbs/docs/tutorials/4_prediction_intervals.ipynb
index 6425dd15b..4c156ed52 100644
--- a/nbs/docs/tutorials/4_prediction_intervals.ipynb
+++ b/nbs/docs/tutorials/4_prediction_intervals.ipynb
@@ -28,7 +28,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -99,7 +99,7 @@
"outputs": [],
"source": [
"import pandas as pd\n",
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/docs/tutorials/5_finetuning.ipynb b/nbs/docs/tutorials/5_finetuning.ipynb
index abb365aa6..7ea4f13d4 100644
--- a/nbs/docs/tutorials/5_finetuning.ipynb
+++ b/nbs/docs/tutorials/5_finetuning.ipynb
@@ -26,7 +26,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -94,7 +94,7 @@
"outputs": [],
"source": [
"import pandas as pd\n",
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/docs/tutorials/6_multiple_series.ipynb b/nbs/docs/tutorials/6_multiple_series.ipynb
index 57b0b280e..0fd488095 100644
--- a/nbs/docs/tutorials/6_multiple_series.ipynb
+++ b/nbs/docs/tutorials/6_multiple_series.ipynb
@@ -24,7 +24,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -92,7 +92,7 @@
"outputs": [],
"source": [
"import pandas as pd\n",
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/docs/tutorials/7_historical_forecast.ipynb b/nbs/docs/tutorials/7_historical_forecast.ipynb
index 64b6b93a9..23d852f4d 100644
--- a/nbs/docs/tutorials/7_historical_forecast.ipynb
+++ b/nbs/docs/tutorials/7_historical_forecast.ipynb
@@ -24,7 +24,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -92,7 +92,7 @@
"outputs": [],
"source": [
"import pandas as pd\n",
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/docs/tutorials/8_irregular_timestamps.ipynb b/nbs/docs/tutorials/8_irregular_timestamps.ipynb
index badf357d8..0e346d4ea 100644
--- a/nbs/docs/tutorials/8_irregular_timestamps.ipynb
+++ b/nbs/docs/tutorials/8_irregular_timestamps.ipynb
@@ -20,7 +20,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -89,7 +89,7 @@
"outputs": [],
"source": [
"import pandas as pd\n",
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/docs/tutorials/9_cross_validation.ipynb b/nbs/docs/tutorials/9_cross_validation.ipynb
index d4c9aa715..a5c671784 100644
--- a/nbs/docs/tutorials/9_cross_validation.ipynb
+++ b/nbs/docs/tutorials/9_cross_validation.ipynb
@@ -26,7 +26,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.utils import colab_badge"
+ "from nixtla.utils import colab_badge"
]
},
{
@@ -71,7 +71,7 @@
"outputs": [],
"source": [
"import pandas as pd\n",
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/index.ipynb b/nbs/index.ipynb
index 75c083187..26cec4461 100644
--- a/nbs/index.ipynb
+++ b/nbs/index.ipynb
@@ -21,7 +21,7 @@
"metadata": {},
"source": [
"```bash\n",
- "pip install nixtlats\n",
+ "pip install nixtla\n",
"```"
]
},
@@ -80,7 +80,7 @@
"metadata": {},
"outputs": [],
"source": [
- "from nixtlats import NixtlaClient"
+ "from nixtla import NixtlaClient"
]
},
{
diff --git a/nbs/nbdev.yml b/nbs/nbdev.yml
index 8bf456cfc..b72003f40 100644
--- a/nbs/nbdev.yml
+++ b/nbs/nbdev.yml
@@ -2,8 +2,8 @@ project:
output-dir: _docs
website:
- title: "nixtlats"
- site-url: "https://Nixtla.github.io/nixtlats/"
+ title: "nixtla"
+ site-url: "https://Nixtla.github.io/nixtla/"
description: "Time series forecasting suite using TimeGPT"
repo-branch: main
repo-url: "https://github.com/Nixtla/nixtla/"
diff --git a/nbs/nixtla_client.ipynb b/nbs/nixtla_client.ipynb
index 714172098..16e82bbdf 100644
--- a/nbs/nixtla_client.ipynb
+++ b/nbs/nixtla_client.ipynb
@@ -67,11 +67,12 @@
" vertical_concat,\n",
")\n",
"\n",
- "from nixtlats.client import (\n",
- " ApiError, Nixtla, \n",
- " SingleSeriesForecast, MultiSeriesForecast, \n",
- " MultiSeriesAnomaly, MultiSeriesInsampleForecast,\n",
- ")\n",
+ "from nixtla.client import Nixtla\n",
+ "from nixtla.core import ApiError\n",
+ "from nixtla.types.multi_series_anomaly import MultiSeriesAnomaly\n",
+ "from nixtla.types.multi_series_forecast import MultiSeriesForecast\n",
+ "from nixtla.types.multi_series_insample_forecast import MultiSeriesInsampleForecast\n",
+ "from nixtla.types.single_series_forecast import SingleSeriesForecast\n",
"\n",
"logging.basicConfig(level=logging.INFO)\n",
"main_logger = logging.getLogger(__name__)\n",
@@ -1177,7 +1178,7 @@
" except ModuleNotFoundError:\n",
" raise Exception(\n",
" 'You have to install additional dependencies to use this method, '\n",
- " 'please install them using `pip install \"nixtlats[plotting]\"`'\n",
+ " 'please install them using `pip install \"nixtla[plotting]\"`'\n",
" )\n",
" df = df.copy()\n",
" if id_col not in df:\n",
@@ -1231,7 +1232,7 @@
"class NixtlaClient(_NixtlaClient):\n",
"\n",
" def _instantiate_distributed_nixtla_client(self):\n",
- " from nixtlats.distributed.nixtla_client import _DistributedNixtlaClient\n",
+ " from nixtla.distributed.nixtla_client import _DistributedNixtlaClient\n",
" dist_nixtla_client = _DistributedNixtlaClient(\n",
" api_key=self.client._client_wrapper._token,\n",
" base_url=self.client._client_wrapper._base_url,\n",
@@ -2144,7 +2145,7 @@
" max_wait_time=max_wait_time,\n",
" )\n",
" init_time = time()\n",
- " with patch('nixtlats.client.Nixtla.forecast_multi_series', side_effect=side_effect):\n",
+ " with patch('nixtla.client.Nixtla.forecast_multi_series', side_effect=side_effect):\n",
" test_fail(\n",
" lambda: mock_nixtla_client.forecast(df=df, h=12, time_col='timestamp', target_col='value'),\n",
" )\n",
@@ -2593,7 +2594,7 @@
"outputs": [],
"source": [
"#| hide\n",
- "from nixtlats.date_features import SpecialDates"
+ "from nixtla.date_features import SpecialDates"
]
},
{
diff --git a/nixtlats/__init__.py b/nixtla/__init__.py
similarity index 76%
rename from nixtlats/__init__.py
rename to nixtla/__init__.py
index ec305560b..da2a85c9b 100644
--- a/nixtlats/__init__.py
+++ b/nixtla/__init__.py
@@ -1,3 +1,3 @@
__version__ = "0.3.0"
-__all__ = ["TimeGPT"]
+
from .nixtla_client import NixtlaClient, TimeGPT
diff --git a/nixtla/_modidx.py b/nixtla/_modidx.py
new file mode 100644
index 000000000..fb3cd855d
--- /dev/null
+++ b/nixtla/_modidx.py
@@ -0,0 +1,161 @@
+# Autogenerated by nbdev
+
+d = { 'settings': { 'branch': 'main',
+ 'doc_baseurl': '/nixtla/',
+ 'doc_host': 'https://Nixtla.github.io',
+ 'git_url': 'https://github.com/Nixtla/nixtla/',
+ 'lib_path': 'nixtla'},
+ 'syms': { 'nixtla.client': {},
+ 'nixtla.core.api_error': {},
+ 'nixtla.core.client_wrapper': {},
+ 'nixtla.core.datetime_utils': {},
+ 'nixtla.core.jsonable_encoder': {},
+ 'nixtla.core.remove_none_from_dict': {},
+ 'nixtla.date_features': { 'nixtla.date_features.CountryHolidays': ( 'date_features.html#countryholidays',
+ 'nixtla/date_features.py'),
+ 'nixtla.date_features.CountryHolidays.__call__': ( 'date_features.html#countryholidays.__call__',
+ 'nixtla/date_features.py'),
+ 'nixtla.date_features.CountryHolidays.__init__': ( 'date_features.html#countryholidays.__init__',
+ 'nixtla/date_features.py'),
+ 'nixtla.date_features.CountryHolidays.__name__': ( 'date_features.html#countryholidays.__name__',
+ 'nixtla/date_features.py'),
+ 'nixtla.date_features.SpecialDates': ('date_features.html#specialdates', 'nixtla/date_features.py'),
+ 'nixtla.date_features.SpecialDates.__call__': ( 'date_features.html#specialdates.__call__',
+ 'nixtla/date_features.py'),
+ 'nixtla.date_features.SpecialDates.__init__': ( 'date_features.html#specialdates.__init__',
+ 'nixtla/date_features.py'),
+ 'nixtla.date_features.SpecialDates.__name__': ( 'date_features.html#specialdates.__name__',
+ 'nixtla/date_features.py'),
+ 'nixtla.date_features._get_holidays_df': ( 'date_features.html#_get_holidays_df',
+ 'nixtla/date_features.py'),
+ 'nixtla.date_features._transform_dict_holidays': ( 'date_features.html#_transform_dict_holidays',
+ 'nixtla/date_features.py')},
+ 'nixtla.distributed.nixtla_client': { 'nixtla.distributed.nixtla_client._DistributedNixtlaClient': ( 'distributed.nixtla_client.html#_distributednixtlaclient',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._DistributedNixtlaClient.__init__': ( 'distributed.nixtla_client.html#_distributednixtlaclient.__init__',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._DistributedNixtlaClient._cross_validation': ( 'distributed.nixtla_client.html#_distributednixtlaclient._cross_validation',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._DistributedNixtlaClient._detect_anomalies': ( 'distributed.nixtla_client.html#_distributednixtlaclient._detect_anomalies',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._DistributedNixtlaClient._distribute_method': ( 'distributed.nixtla_client.html#_distributednixtlaclient._distribute_method',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._DistributedNixtlaClient._forecast': ( 'distributed.nixtla_client.html#_distributednixtlaclient._forecast',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._DistributedNixtlaClient._forecast_x': ( 'distributed.nixtla_client.html#_distributednixtlaclient._forecast_x',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._DistributedNixtlaClient._get_anomalies_schema': ( 'distributed.nixtla_client.html#_distributednixtlaclient._get_anomalies_schema',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._DistributedNixtlaClient._get_forecast_schema': ( 'distributed.nixtla_client.html#_distributednixtlaclient._get_forecast_schema',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._DistributedNixtlaClient._instantiate_nixtla_client': ( 'distributed.nixtla_client.html#_distributednixtlaclient._instantiate_nixtla_client',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._DistributedNixtlaClient.cross_validation': ( 'distributed.nixtla_client.html#_distributednixtlaclient.cross_validation',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._DistributedNixtlaClient.detect_anomalies': ( 'distributed.nixtla_client.html#_distributednixtlaclient.detect_anomalies',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._DistributedNixtlaClient.forecast': ( 'distributed.nixtla_client.html#_distributednixtlaclient.forecast',
+ 'nixtla/distributed/nixtla_client.py'),
+ 'nixtla.distributed.nixtla_client._cotransform': ( 'distributed.nixtla_client.html#_cotransform',
+ 'nixtla/distributed/nixtla_client.py')},
+ 'nixtla.errors.unprocessable_entity_error': {},
+ 'nixtla.nixtla_client': { 'nixtla.nixtla_client.NixtlaClient': ('nixtla_client.html#nixtlaclient', 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client.NixtlaClient._instantiate_distributed_nixtla_client': ( 'nixtla_client.html#nixtlaclient._instantiate_distributed_nixtla_client',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client.NixtlaClient.cross_validation': ( 'nixtla_client.html#nixtlaclient.cross_validation',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client.NixtlaClient.detect_anomalies': ( 'nixtla_client.html#nixtlaclient.detect_anomalies',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client.NixtlaClient.forecast': ( 'nixtla_client.html#nixtlaclient.forecast',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client.TimeGPT': ('nixtla_client.html#timegpt', 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client.TimeGPT.__init__': ( 'nixtla_client.html#timegpt.__init__',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClient': ('nixtla_client.html#_nixtlaclient', 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClient.__init__': ( 'nixtla_client.html#_nixtlaclient.__init__',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClient._cross_validation': ( 'nixtla_client.html#_nixtlaclient._cross_validation',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClient._detect_anomalies': ( 'nixtla_client.html#_nixtlaclient._detect_anomalies',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClient._forecast': ( 'nixtla_client.html#_nixtlaclient._forecast',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClient.plot': ( 'nixtla_client.html#_nixtlaclient.plot',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClient.validate_api_key': ( 'nixtla_client.html#_nixtlaclient.validate_api_key',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClient.validate_token': ( 'nixtla_client.html#_nixtlaclient.validate_token',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel': ( 'nixtla_client.html#_nixtlaclientmodel',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.__init__': ( 'nixtla_client.html#_nixtlaclientmodel.__init__',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel._call_api': ( 'nixtla_client.html#_nixtlaclientmodel._call_api',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel._prepare_level_and_quantiles': ( 'nixtla_client.html#_nixtlaclientmodel._prepare_level_and_quantiles',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel._retry_strategy': ( 'nixtla_client.html#_nixtlaclientmodel._retry_strategy',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.add_date_features': ( 'nixtla_client.html#_nixtlaclientmodel.add_date_features',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.compute_date_feature': ( 'nixtla_client.html#_nixtlaclientmodel.compute_date_feature',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.cross_validation': ( 'nixtla_client.html#_nixtlaclientmodel.cross_validation',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.dataframes_to_dict': ( 'nixtla_client.html#_nixtlaclientmodel.dataframes_to_dict',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.detect_anomalies': ( 'nixtla_client.html#_nixtlaclientmodel.detect_anomalies',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.forecast': ( 'nixtla_client.html#_nixtlaclientmodel.forecast',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.infer_freq': ( 'nixtla_client.html#_nixtlaclientmodel.infer_freq',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.make_future_dataframe': ( 'nixtla_client.html#_nixtlaclientmodel.make_future_dataframe',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.preprocess_X_df': ( 'nixtla_client.html#_nixtlaclientmodel.preprocess_x_df',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.preprocess_dataframes': ( 'nixtla_client.html#_nixtlaclientmodel.preprocess_dataframes',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.resample_dataframe': ( 'nixtla_client.html#_nixtlaclientmodel.resample_dataframe',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.set_model_params': ( 'nixtla_client.html#_nixtlaclientmodel.set_model_params',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.transform_inputs': ( 'nixtla_client.html#_nixtlaclientmodel.transform_inputs',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.transform_outputs': ( 'nixtla_client.html#_nixtlaclientmodel.transform_outputs',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client._NixtlaClientModel.validate_input_size': ( 'nixtla_client.html#_nixtlaclientmodel.validate_input_size',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client.deprecated_argument': ( 'nixtla_client.html#deprecated_argument',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client.deprecated_method': ( 'nixtla_client.html#deprecated_method',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client.partition_by_uid': ( 'nixtla_client.html#partition_by_uid',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client.remove_unused_categories': ( 'nixtla_client.html#remove_unused_categories',
+ 'nixtla/nixtla_client.py'),
+ 'nixtla.nixtla_client.validate_model_parameter': ( 'nixtla_client.html#validate_model_parameter',
+ 'nixtla/nixtla_client.py')},
+ 'nixtla.types.http_validation_error': {},
+ 'nixtla.types.multi_series_anomaly': {},
+ 'nixtla.types.multi_series_anomaly_model': {},
+ 'nixtla.types.multi_series_cross_validation': {},
+ 'nixtla.types.multi_series_cross_validation_fewshot_loss': {},
+ 'nixtla.types.multi_series_cross_validation_finetune_loss': {},
+ 'nixtla.types.multi_series_cross_validation_model': {},
+ 'nixtla.types.multi_series_forecast': {},
+ 'nixtla.types.multi_series_forecast_fewshot_loss': {},
+ 'nixtla.types.multi_series_forecast_finetune_loss': {},
+ 'nixtla.types.multi_series_forecast_model': {},
+ 'nixtla.types.multi_series_input': {},
+ 'nixtla.types.multi_series_insample_forecast': {},
+ 'nixtla.types.multi_series_insample_forecast_model': {},
+ 'nixtla.types.single_series_forecast': {},
+ 'nixtla.types.single_series_forecast_fewshot_loss': {},
+ 'nixtla.types.single_series_forecast_finetune_loss': {},
+ 'nixtla.types.single_series_forecast_model': {},
+ 'nixtla.types.single_series_insample_forecast': {},
+ 'nixtla.types.single_series_insample_forecast_model': {},
+ 'nixtla.types.validation_error': {},
+ 'nixtla.types.validation_error_loc_item': {},
+ 'nixtla.utils': {'nixtla.utils.colab_badge': ('utils.html#colab_badge', 'nixtla/utils.py')}}}
diff --git a/nixtlats/client.py b/nixtla/client.py
similarity index 100%
rename from nixtlats/client.py
rename to nixtla/client.py
diff --git a/nixtlats/core/__init__.py b/nixtla/core/__init__.py
similarity index 100%
rename from nixtlats/core/__init__.py
rename to nixtla/core/__init__.py
diff --git a/nixtlats/core/api_error.py b/nixtla/core/api_error.py
similarity index 100%
rename from nixtlats/core/api_error.py
rename to nixtla/core/api_error.py
diff --git a/nixtlats/core/client_wrapper.py b/nixtla/core/client_wrapper.py
similarity index 100%
rename from nixtlats/core/client_wrapper.py
rename to nixtla/core/client_wrapper.py
diff --git a/nixtlats/core/datetime_utils.py b/nixtla/core/datetime_utils.py
similarity index 100%
rename from nixtlats/core/datetime_utils.py
rename to nixtla/core/datetime_utils.py
diff --git a/nixtlats/core/jsonable_encoder.py b/nixtla/core/jsonable_encoder.py
similarity index 100%
rename from nixtlats/core/jsonable_encoder.py
rename to nixtla/core/jsonable_encoder.py
diff --git a/nixtlats/core/remove_none_from_dict.py b/nixtla/core/remove_none_from_dict.py
similarity index 100%
rename from nixtlats/core/remove_none_from_dict.py
rename to nixtla/core/remove_none_from_dict.py
diff --git a/nixtlats/date_features.py b/nixtla/date_features.py
similarity index 96%
rename from nixtlats/date_features.py
rename to nixtla/date_features.py
index d52843fe4..d3e4025c5 100644
--- a/nixtlats/date_features.py
+++ b/nixtla/date_features.py
@@ -43,7 +43,7 @@ def __call__(self, dates: pd.DatetimeIndex):
except ModuleNotFoundError:
raise Exception(
"You have to install additional libraries to use holidays, "
- 'please install them using `pip install "nixtlats[date_extras]"`'
+ 'please install them using `pip install "nixtla[date_extras]"`'
)
return _get_holidays_df(
dates, self.countries, country_holidays, list_supported_countries()
diff --git a/nixtlats/distributed/__init__.py b/nixtla/distributed/__init__.py
similarity index 100%
rename from nixtlats/distributed/__init__.py
rename to nixtla/distributed/__init__.py
diff --git a/nixtlats/distributed/nixtla_client.py b/nixtla/distributed/nixtla_client.py
similarity index 99%
rename from nixtlats/distributed/nixtla_client.py
rename to nixtla/distributed/nixtla_client.py
index 172bfac7e..7160c857a 100644
--- a/nixtlats/distributed/nixtla_client.py
+++ b/nixtla/distributed/nixtla_client.py
@@ -266,7 +266,7 @@ def cross_validation(
return fcst_df
def _instantiate_nixtla_client(self):
- from nixtlats.nixtla_client import _NixtlaClient
+ from nixtla.nixtla_client import _NixtlaClient
nixtla_client = _NixtlaClient(
api_key=self.api_key,
diff --git a/nixtlats/errors/__init__.py b/nixtla/errors/__init__.py
similarity index 100%
rename from nixtlats/errors/__init__.py
rename to nixtla/errors/__init__.py
diff --git a/nixtlats/errors/unprocessable_entity_error.py b/nixtla/errors/unprocessable_entity_error.py
similarity index 100%
rename from nixtlats/errors/unprocessable_entity_error.py
rename to nixtla/errors/unprocessable_entity_error.py
diff --git a/nixtlats/nixtla_client.py b/nixtla/nixtla_client.py
similarity index 99%
rename from nixtlats/nixtla_client.py
rename to nixtla/nixtla_client.py
index 6b9ffe552..dfc9f9651 100644
--- a/nixtlats/nixtla_client.py
+++ b/nixtla/nixtla_client.py
@@ -39,14 +39,12 @@
vertical_concat,
)
-from nixtlats.client import (
- ApiError,
- Nixtla,
- SingleSeriesForecast,
- MultiSeriesForecast,
- MultiSeriesAnomaly,
- MultiSeriesInsampleForecast,
-)
+from .client import Nixtla
+from .core import ApiError
+from .types.multi_series_anomaly import MultiSeriesAnomaly
+from .types.multi_series_forecast import MultiSeriesForecast
+from .types.multi_series_insample_forecast import MultiSeriesInsampleForecast
+from .types.single_series_forecast import SingleSeriesForecast
logging.basicConfig(level=logging.INFO)
main_logger = logging.getLogger(__name__)
@@ -1113,7 +1111,7 @@ def plot(
except ModuleNotFoundError:
raise Exception(
"You have to install additional dependencies to use this method, "
- 'please install them using `pip install "nixtlats[plotting]"`'
+ 'please install them using `pip install "nixtla[plotting]"`'
)
df = df.copy()
if id_col not in df:
@@ -1160,7 +1158,7 @@ def plot(
class NixtlaClient(_NixtlaClient):
def _instantiate_distributed_nixtla_client(self):
- from nixtlats.distributed.nixtla_client import _DistributedNixtlaClient
+ from nixtla.distributed.nixtla_client import _DistributedNixtlaClient
dist_nixtla_client = _DistributedNixtlaClient(
api_key=self.client._client_wrapper._token,
diff --git a/nixtlats/py.typed b/nixtla/py.typed
similarity index 100%
rename from nixtlats/py.typed
rename to nixtla/py.typed
diff --git a/nixtlats/types/__init__.py b/nixtla/types/__init__.py
similarity index 100%
rename from nixtlats/types/__init__.py
rename to nixtla/types/__init__.py
diff --git a/nixtlats/types/http_validation_error.py b/nixtla/types/http_validation_error.py
similarity index 100%
rename from nixtlats/types/http_validation_error.py
rename to nixtla/types/http_validation_error.py
diff --git a/nixtlats/types/multi_series_anomaly.py b/nixtla/types/multi_series_anomaly.py
similarity index 100%
rename from nixtlats/types/multi_series_anomaly.py
rename to nixtla/types/multi_series_anomaly.py
diff --git a/nixtlats/types/multi_series_anomaly_model.py b/nixtla/types/multi_series_anomaly_model.py
similarity index 100%
rename from nixtlats/types/multi_series_anomaly_model.py
rename to nixtla/types/multi_series_anomaly_model.py
diff --git a/nixtlats/types/multi_series_cross_validation.py b/nixtla/types/multi_series_cross_validation.py
similarity index 100%
rename from nixtlats/types/multi_series_cross_validation.py
rename to nixtla/types/multi_series_cross_validation.py
diff --git a/nixtlats/types/multi_series_cross_validation_fewshot_loss.py b/nixtla/types/multi_series_cross_validation_fewshot_loss.py
similarity index 100%
rename from nixtlats/types/multi_series_cross_validation_fewshot_loss.py
rename to nixtla/types/multi_series_cross_validation_fewshot_loss.py
diff --git a/nixtlats/types/multi_series_cross_validation_finetune_loss.py b/nixtla/types/multi_series_cross_validation_finetune_loss.py
similarity index 100%
rename from nixtlats/types/multi_series_cross_validation_finetune_loss.py
rename to nixtla/types/multi_series_cross_validation_finetune_loss.py
diff --git a/nixtlats/types/multi_series_cross_validation_model.py b/nixtla/types/multi_series_cross_validation_model.py
similarity index 100%
rename from nixtlats/types/multi_series_cross_validation_model.py
rename to nixtla/types/multi_series_cross_validation_model.py
diff --git a/nixtlats/types/multi_series_forecast.py b/nixtla/types/multi_series_forecast.py
similarity index 100%
rename from nixtlats/types/multi_series_forecast.py
rename to nixtla/types/multi_series_forecast.py
diff --git a/nixtlats/types/multi_series_forecast_fewshot_loss.py b/nixtla/types/multi_series_forecast_fewshot_loss.py
similarity index 100%
rename from nixtlats/types/multi_series_forecast_fewshot_loss.py
rename to nixtla/types/multi_series_forecast_fewshot_loss.py
diff --git a/nixtlats/types/multi_series_forecast_finetune_loss.py b/nixtla/types/multi_series_forecast_finetune_loss.py
similarity index 100%
rename from nixtlats/types/multi_series_forecast_finetune_loss.py
rename to nixtla/types/multi_series_forecast_finetune_loss.py
diff --git a/nixtlats/types/multi_series_forecast_model.py b/nixtla/types/multi_series_forecast_model.py
similarity index 100%
rename from nixtlats/types/multi_series_forecast_model.py
rename to nixtla/types/multi_series_forecast_model.py
diff --git a/nixtlats/types/multi_series_input.py b/nixtla/types/multi_series_input.py
similarity index 100%
rename from nixtlats/types/multi_series_input.py
rename to nixtla/types/multi_series_input.py
diff --git a/nixtlats/types/multi_series_insample_forecast.py b/nixtla/types/multi_series_insample_forecast.py
similarity index 100%
rename from nixtlats/types/multi_series_insample_forecast.py
rename to nixtla/types/multi_series_insample_forecast.py
diff --git a/nixtlats/types/multi_series_insample_forecast_model.py b/nixtla/types/multi_series_insample_forecast_model.py
similarity index 100%
rename from nixtlats/types/multi_series_insample_forecast_model.py
rename to nixtla/types/multi_series_insample_forecast_model.py
diff --git a/nixtlats/types/single_series_forecast.py b/nixtla/types/single_series_forecast.py
similarity index 100%
rename from nixtlats/types/single_series_forecast.py
rename to nixtla/types/single_series_forecast.py
diff --git a/nixtlats/types/single_series_forecast_fewshot_loss.py b/nixtla/types/single_series_forecast_fewshot_loss.py
similarity index 100%
rename from nixtlats/types/single_series_forecast_fewshot_loss.py
rename to nixtla/types/single_series_forecast_fewshot_loss.py
diff --git a/nixtlats/types/single_series_forecast_finetune_loss.py b/nixtla/types/single_series_forecast_finetune_loss.py
similarity index 100%
rename from nixtlats/types/single_series_forecast_finetune_loss.py
rename to nixtla/types/single_series_forecast_finetune_loss.py
diff --git a/nixtlats/types/single_series_forecast_model.py b/nixtla/types/single_series_forecast_model.py
similarity index 100%
rename from nixtlats/types/single_series_forecast_model.py
rename to nixtla/types/single_series_forecast_model.py
diff --git a/nixtlats/types/single_series_insample_forecast.py b/nixtla/types/single_series_insample_forecast.py
similarity index 100%
rename from nixtlats/types/single_series_insample_forecast.py
rename to nixtla/types/single_series_insample_forecast.py
diff --git a/nixtlats/types/single_series_insample_forecast_model.py b/nixtla/types/single_series_insample_forecast_model.py
similarity index 100%
rename from nixtlats/types/single_series_insample_forecast_model.py
rename to nixtla/types/single_series_insample_forecast_model.py
diff --git a/nixtlats/types/validation_error.py b/nixtla/types/validation_error.py
similarity index 100%
rename from nixtlats/types/validation_error.py
rename to nixtla/types/validation_error.py
diff --git a/nixtlats/types/validation_error_loc_item.py b/nixtla/types/validation_error_loc_item.py
similarity index 100%
rename from nixtlats/types/validation_error_loc_item.py
rename to nixtla/types/validation_error_loc_item.py
diff --git a/nixtlats/utils.py b/nixtla/utils.py
similarity index 100%
rename from nixtlats/utils.py
rename to nixtla/utils.py
diff --git a/nixtlats/_modidx.py b/nixtlats/_modidx.py
deleted file mode 100644
index d76a93899..000000000
--- a/nixtlats/_modidx.py
+++ /dev/null
@@ -1,164 +0,0 @@
-# Autogenerated by nbdev
-
-d = { 'settings': { 'branch': 'main',
- 'doc_baseurl': '/nixtlats/',
- 'doc_host': 'https://Nixtla.github.io',
- 'git_url': 'https://github.com/Nixtla/nixtla/',
- 'lib_path': 'nixtlats'},
- 'syms': { 'nixtlats.client': {},
- 'nixtlats.core.api_error': {},
- 'nixtlats.core.client_wrapper': {},
- 'nixtlats.core.datetime_utils': {},
- 'nixtlats.core.jsonable_encoder': {},
- 'nixtlats.core.remove_none_from_dict': {},
- 'nixtlats.date_features': { 'nixtlats.date_features.CountryHolidays': ( 'date_features.html#countryholidays',
- 'nixtlats/date_features.py'),
- 'nixtlats.date_features.CountryHolidays.__call__': ( 'date_features.html#countryholidays.__call__',
- 'nixtlats/date_features.py'),
- 'nixtlats.date_features.CountryHolidays.__init__': ( 'date_features.html#countryholidays.__init__',
- 'nixtlats/date_features.py'),
- 'nixtlats.date_features.CountryHolidays.__name__': ( 'date_features.html#countryholidays.__name__',
- 'nixtlats/date_features.py'),
- 'nixtlats.date_features.SpecialDates': ( 'date_features.html#specialdates',
- 'nixtlats/date_features.py'),
- 'nixtlats.date_features.SpecialDates.__call__': ( 'date_features.html#specialdates.__call__',
- 'nixtlats/date_features.py'),
- 'nixtlats.date_features.SpecialDates.__init__': ( 'date_features.html#specialdates.__init__',
- 'nixtlats/date_features.py'),
- 'nixtlats.date_features.SpecialDates.__name__': ( 'date_features.html#specialdates.__name__',
- 'nixtlats/date_features.py'),
- 'nixtlats.date_features._get_holidays_df': ( 'date_features.html#_get_holidays_df',
- 'nixtlats/date_features.py'),
- 'nixtlats.date_features._transform_dict_holidays': ( 'date_features.html#_transform_dict_holidays',
- 'nixtlats/date_features.py')},
- 'nixtlats.distributed.nixtla_client': { 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient': ( 'distributed.nixtla_client.html#_distributednixtlaclient',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient.__init__': ( 'distributed.nixtla_client.html#_distributednixtlaclient.__init__',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient._cross_validation': ( 'distributed.nixtla_client.html#_distributednixtlaclient._cross_validation',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient._detect_anomalies': ( 'distributed.nixtla_client.html#_distributednixtlaclient._detect_anomalies',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient._distribute_method': ( 'distributed.nixtla_client.html#_distributednixtlaclient._distribute_method',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient._forecast': ( 'distributed.nixtla_client.html#_distributednixtlaclient._forecast',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient._forecast_x': ( 'distributed.nixtla_client.html#_distributednixtlaclient._forecast_x',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient._get_anomalies_schema': ( 'distributed.nixtla_client.html#_distributednixtlaclient._get_anomalies_schema',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient._get_forecast_schema': ( 'distributed.nixtla_client.html#_distributednixtlaclient._get_forecast_schema',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient._instantiate_nixtla_client': ( 'distributed.nixtla_client.html#_distributednixtlaclient._instantiate_nixtla_client',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient.cross_validation': ( 'distributed.nixtla_client.html#_distributednixtlaclient.cross_validation',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient.detect_anomalies': ( 'distributed.nixtla_client.html#_distributednixtlaclient.detect_anomalies',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._DistributedNixtlaClient.forecast': ( 'distributed.nixtla_client.html#_distributednixtlaclient.forecast',
- 'nixtlats/distributed/nixtla_client.py'),
- 'nixtlats.distributed.nixtla_client._cotransform': ( 'distributed.nixtla_client.html#_cotransform',
- 'nixtlats/distributed/nixtla_client.py')},
- 'nixtlats.errors.unprocessable_entity_error': {},
- 'nixtlats.nixtla_client': { 'nixtlats.nixtla_client.NixtlaClient': ( 'nixtla_client.html#nixtlaclient',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client.NixtlaClient._instantiate_distributed_nixtla_client': ( 'nixtla_client.html#nixtlaclient._instantiate_distributed_nixtla_client',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client.NixtlaClient.cross_validation': ( 'nixtla_client.html#nixtlaclient.cross_validation',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client.NixtlaClient.detect_anomalies': ( 'nixtla_client.html#nixtlaclient.detect_anomalies',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client.NixtlaClient.forecast': ( 'nixtla_client.html#nixtlaclient.forecast',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client.TimeGPT': ('nixtla_client.html#timegpt', 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client.TimeGPT.__init__': ( 'nixtla_client.html#timegpt.__init__',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClient': ( 'nixtla_client.html#_nixtlaclient',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClient.__init__': ( 'nixtla_client.html#_nixtlaclient.__init__',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClient._cross_validation': ( 'nixtla_client.html#_nixtlaclient._cross_validation',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClient._detect_anomalies': ( 'nixtla_client.html#_nixtlaclient._detect_anomalies',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClient._forecast': ( 'nixtla_client.html#_nixtlaclient._forecast',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClient.plot': ( 'nixtla_client.html#_nixtlaclient.plot',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClient.validate_api_key': ( 'nixtla_client.html#_nixtlaclient.validate_api_key',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClient.validate_token': ( 'nixtla_client.html#_nixtlaclient.validate_token',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel': ( 'nixtla_client.html#_nixtlaclientmodel',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.__init__': ( 'nixtla_client.html#_nixtlaclientmodel.__init__',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel._call_api': ( 'nixtla_client.html#_nixtlaclientmodel._call_api',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel._prepare_level_and_quantiles': ( 'nixtla_client.html#_nixtlaclientmodel._prepare_level_and_quantiles',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel._retry_strategy': ( 'nixtla_client.html#_nixtlaclientmodel._retry_strategy',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.add_date_features': ( 'nixtla_client.html#_nixtlaclientmodel.add_date_features',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.compute_date_feature': ( 'nixtla_client.html#_nixtlaclientmodel.compute_date_feature',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.cross_validation': ( 'nixtla_client.html#_nixtlaclientmodel.cross_validation',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.dataframes_to_dict': ( 'nixtla_client.html#_nixtlaclientmodel.dataframes_to_dict',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.detect_anomalies': ( 'nixtla_client.html#_nixtlaclientmodel.detect_anomalies',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.forecast': ( 'nixtla_client.html#_nixtlaclientmodel.forecast',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.infer_freq': ( 'nixtla_client.html#_nixtlaclientmodel.infer_freq',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.make_future_dataframe': ( 'nixtla_client.html#_nixtlaclientmodel.make_future_dataframe',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.preprocess_X_df': ( 'nixtla_client.html#_nixtlaclientmodel.preprocess_x_df',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.preprocess_dataframes': ( 'nixtla_client.html#_nixtlaclientmodel.preprocess_dataframes',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.resample_dataframe': ( 'nixtla_client.html#_nixtlaclientmodel.resample_dataframe',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.set_model_params': ( 'nixtla_client.html#_nixtlaclientmodel.set_model_params',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.transform_inputs': ( 'nixtla_client.html#_nixtlaclientmodel.transform_inputs',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.transform_outputs': ( 'nixtla_client.html#_nixtlaclientmodel.transform_outputs',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client._NixtlaClientModel.validate_input_size': ( 'nixtla_client.html#_nixtlaclientmodel.validate_input_size',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client.deprecated_argument': ( 'nixtla_client.html#deprecated_argument',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client.deprecated_method': ( 'nixtla_client.html#deprecated_method',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client.partition_by_uid': ( 'nixtla_client.html#partition_by_uid',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client.remove_unused_categories': ( 'nixtla_client.html#remove_unused_categories',
- 'nixtlats/nixtla_client.py'),
- 'nixtlats.nixtla_client.validate_model_parameter': ( 'nixtla_client.html#validate_model_parameter',
- 'nixtlats/nixtla_client.py')},
- 'nixtlats.types.http_validation_error': {},
- 'nixtlats.types.multi_series_anomaly': {},
- 'nixtlats.types.multi_series_anomaly_model': {},
- 'nixtlats.types.multi_series_cross_validation': {},
- 'nixtlats.types.multi_series_cross_validation_fewshot_loss': {},
- 'nixtlats.types.multi_series_cross_validation_finetune_loss': {},
- 'nixtlats.types.multi_series_cross_validation_model': {},
- 'nixtlats.types.multi_series_forecast': {},
- 'nixtlats.types.multi_series_forecast_fewshot_loss': {},
- 'nixtlats.types.multi_series_forecast_finetune_loss': {},
- 'nixtlats.types.multi_series_forecast_model': {},
- 'nixtlats.types.multi_series_input': {},
- 'nixtlats.types.multi_series_insample_forecast': {},
- 'nixtlats.types.multi_series_insample_forecast_model': {},
- 'nixtlats.types.single_series_forecast': {},
- 'nixtlats.types.single_series_forecast_fewshot_loss': {},
- 'nixtlats.types.single_series_forecast_finetune_loss': {},
- 'nixtlats.types.single_series_forecast_model': {},
- 'nixtlats.types.single_series_insample_forecast': {},
- 'nixtlats.types.single_series_insample_forecast_model': {},
- 'nixtlats.types.validation_error': {},
- 'nixtlats.types.validation_error_loc_item': {},
- 'nixtlats.utils': {'nixtlats.utils.colab_badge': ('utils.html#colab_badge', 'nixtlats/utils.py')}}}
diff --git a/settings.ini b/settings.ini
index 2b986f423..dbc9518ab 100644
--- a/settings.ini
+++ b/settings.ini
@@ -1,6 +1,6 @@
[DEFAULT]
host = github
-lib_name = nixtlats
+lib_name = nixtla
user = Nixtla
description = Time series forecasting suite using TimeGPT
keywords = time-series forecasting gpt
diff --git a/setup.py b/setup.py
index cf6ef704b..3ea0b5f43 100644
--- a/setup.py
+++ b/setup.py
@@ -17,19 +17,19 @@
date_extras = ["holidays"]
setuptools.setup(
- name="nixtlats",
+ name="nixtla",
version="0.3.0",
description="Python SDK for Nixtla API (TimeGPT)",
long_description=long_description,
long_description_content_type="text/markdown",
url="https://github.com/Nixtla/nixtla",
- packages=setuptools.find_packages(),
+ packages=setuptools.find_packages(exclude=["action_files"]),
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
- python_requires=">=3.7",
+ python_requires=">=3.8",
install_requires=[
"httpx",
"pandas",