Make pymc optional (#42)

* make pymc optional * up version and add test dep * accidental file
wd60622 · Jan 22, 2024 · 9bfddd2 · 9bfddd2
1 parent 8dbe4a6
commit 9bfddd2
Show file tree

Hide file tree

Showing 3 changed files with 645 additions and 696 deletions.
diff --git a/latent_calendar/generate.py b/latent_calendar/generate.py
@@ -4,11 +4,27 @@
 import numpy as np
 import pandas as pd
 
-import pymc as pm
-from pytensor.tensor import TensorVariable
-
 from latent_calendar.const import FULL_VOCAB
 
+try:
+    import pymc as pm
+    from pytensor.tensor import TensorVariable
+except ImportError:
+
+    class TensorVariable:
+        pass
+
+    class PyMC:
+        def __getattr__(self, name):
+            msg = (
+                "PyMC is not installed."
+                " Please install it directly or"
+                " with extra installs: pip install 'latent-calendar[pymc]'"
+            )
+            raise ImportError(msg)
+
+    pm = PyMC()
+
 
 def wide_format_dataframe(
     n_rows: int, rate: float = 1.0, random_state: Optional[int] = None