Create a minimal environment with core dependencies

README.md

@@ -56,8 +56,13 @@ You can install the release version from `PyPI` by running
 # inside an active virtual environment
 python -m pip install pyfixest
 ```
+To install a version with only core functionality and a reduced dependency set, you can run
 
-or the development version from github by running
+```py
+python -m pip install pyfixest[minimal]
+```
+
+You can install the development version from github by running
 
 ```py
 python -m pip install git+https://github.com/py-econometrics/pyfixest

pyfixest/estimation/decomposition.py

@@ -3,10 +3,24 @@
 
 import numpy as np
 import pandas as pd
-from joblib import Parallel, delayed
 from scipy.sparse import csc_matrix
 from scipy.sparse.linalg import lsqr
-from tqdm import tqdm
+
+try:
+    from tqdm import tqdm
+except ImportError:
+
+    def tqdm(iterable, *args, **kwargs):
+        "Define a dummy tqdm function."
+        return iterable
+
+
+try:
+    from joblib import Parallel, delayed
+
+    joblib_available = True
+except ImportError:
+    joblib_available = False
 
 
 @dataclass
@@ -154,9 +168,12 @@
         self.alpha = alpha
         self.B = B
 
-        _bootstrapped = Parallel(n_jobs=self.nthreads)(
-            delayed(self._bootstrap)(rng=rng) for _ in tqdm(range(B))
-        )
+        if joblib_available:
+            _bootstrapped = Parallel(n_jobs=self.nthreads)(
+                delayed(self._bootstrap)(rng=rng) for _ in tqdm(range(B))
+            )
+        else:
+            _bootstrapped = [self._bootstrap(rng=rng) for _ in tqdm(range(B))]
 
         self._bootstrapped = {
             key: np.concatenate([d[key] for d in _bootstrapped])

pyfixest/estimation/feols_compressed_.py

@@ -6,7 +6,15 @@
 import narwhals as nw
 import numpy as np
 import pandas as pd
-from tqdm import tqdm
+
+try:
+    from tqdm import tqdm
+except ImportError:
+
+    def tqdm(iterable, *args, **kwargs):
+        "Define a dummy tqdm function."
+        return iterable
+
 
 from pyfixest.estimation.feols_ import Feols, PredictionErrorOptions, PredictionType
 from pyfixest.estimation.FormulaParser import FixestFormula

pyfixest/estimation/ritest.py

@@ -26,7 +26,15 @@
     _HAS_LETS_PLOT = False
 
 from scipy.stats import norm
-from tqdm import tqdm
+
+try:
+    from tqdm import tqdm
+except ImportError:
+
+    def tqdm(iterable, *args, **kwargs):
+        """Define a dummy tqdm function."""
+        return iterable
+
 
 from pyfixest.estimation.demean_ import demean
 

pyfixest/report/summarize.py

@@ -6,8 +6,6 @@
 
 import numpy as np
 import pandas as pd
-from great_tables import GT
-from tabulate import tabulate
 
 from pyfixest.estimation.feiv_ import Feiv
 from pyfixest.estimation.feols_ import Feols
@@ -156,6 +154,8 @@
     pf.etable([fit1, fit2])
     ```
     """
+    # check if etable installed else error
+
     if signif_code is None:
         signif_code = [0.001, 0.01, 0.05]
     assert isinstance(signif_code, list) and len(signif_code) == 3, (
@@ -683,6 +683,11 @@
     -------
     - formatted_table (str): The formatted table as a string.
     """
+    try:
+        from tabulate import tabulate
+    except ImportError:
+        raise ImportError("The tabulate package is required but it is not installed.")
+
     # Format the DataFrame for tabulate
     table = tabulate(
         df,
@@ -1064,6 +1069,13 @@
             rowname_col = dfs.columns[0]
             groupname_col = None
 
+        try:
+            from great_tables import GT
+        except ImportError:
+            raise ImportError(
+                "The great_tables package is for running pf.etable() in 'gt' mode."
+            )
+
         # Generate the table with GT
         gt = GT(dfs, auto_align=False)
 

pyproject.toml

@@ -15,19 +15,23 @@ dependencies = [
   "formulaic>=1.1.0",
   "pandas>=1.1.0",
   "numba>=0.58.0",
-  "seaborn>=0.13.2",
-  "tabulate>=0.9.0",
-  "tqdm>=4.0.0",
-  "great-tables>=0.10.0",
   "numpy>=1.19.0",
   "narwhals>=1.13.3",
-  "joblib>=1.4.2,<2",
+  "seaborn>=0.13.2",
 ]
 
 [tool.pixi.feature.dev.dependencies]
 rpy2 = ">=3.5.11,<4"
 
 [project.optional-dependencies]
+
+core = [
+  "joblib>=1.4.2,<2",
+  "tabulate>=0.9.0",
+  "tqdm>=4.0.0",
+  "great-tables>=0.10.0",
+]
+
 dev = [
   "pytest>=7.2.0",
   "pytest-cov>=4.1.0",
@@ -83,7 +87,8 @@ platforms = ["linux-64", "win-64", "osx-arm64", "osx-64"]
 pyfixest = { path = ".", editable = true }
 
 [tool.pixi.environments]
-default = { solve-group = "default" }
+minimal = { solve-group = "default" }
+default = {features = ["core"], solve-group = "default"}
 dev = { features = ["dev"], solve-group = "default" }
 docs = { features = ["docs"], solve-group = "default" }
 build = { features = ["build"], solve-group = "default" }

scripts/run_notebooks.py

@@ -8,7 +8,14 @@
 
 import papermill
 from joblib import Parallel, delayed
-from tqdm import tqdm
+
+try:
+    from tqdm import tqdm
+except ImportError:
+
+    def tqdm(iterable, *args, **kwargs):
+        return iterable
+
 
 KERNEL_NAME: str = "python3"
 DOCS = Path("docs")