fix: fix package name confusion (arrow <-> pyarrow)

datamole-ai · Oct 10, 2023 · be9d70b · be9d70b
1 parent 4a83b03
commit be9d70b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/edvart/data_types.py b/edvart/data_types.py
@@ -4,7 +4,7 @@
 import pandas as pd
 
 try:
-    import pyarrow  # pylint: disable=unused-import
+    import arrow  # pylint: disable=unused-import
 except ImportError:
     PYARROW_PANDAS_BACKEND_AVAILABLE = False
 else:

diff --git a/pyproject.toml b/pyproject.toml
@@ -33,7 +33,7 @@ pyarrow = { version = "^13.0.0", optional = true }
 [tool.poetry.extras]
 umap = ["umap-learn", "numba"]
 arrow = ["pyarrow"]
-all = ["umap-learn", "numba", "arrow"]
+all = ["umap-learn", "numba", "pyarrow"]
 
 [tool.poetry.dev-dependencies]
 pytest-cov = "^2.8"