Correcting error on importing

pymc-labs · Oct 23, 2023 · 13b6e96 · 13b6e96
1 parent e2a5eb6
commit 13b6e96
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/pymc_marketing/mmm/base.py b/pymc_marketing/mmm/base.py
@@ -7,6 +7,7 @@
     ismethoddescriptor,
 )
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
+import types
 
 import arviz as az
 import matplotlib.pyplot as plt

diff --git a/pymc_marketing/mmm/delayed_saturated_mmm.py b/pymc_marketing/mmm/delayed_saturated_mmm.py
@@ -1,6 +1,5 @@
 import json
 from pathlib import Path
-import types
 from typing import Any, Dict, List, Optional, Union
 
 import arviz as az