ModuleNotFoundError: No module named 'econml.tests'

I tried to run this on my local and Google Colab, both provided the same error.

I am simply trying to run an example in your documentation:

# Main imports
from econml.dynamic.dml import DynamicDML
from econml.tests.dgp import DynamicPanelDGP, add_vlines

# Helper imports
import numpy as np
from sklearn.linear_model import Lasso, LassoCV, LogisticRegression, LogisticRegressionCV, MultiTaskLassoCV
import matplotlib.pyplot as plt

%matplotlib inline

Traceback below:

ModuleNotFoundError                       Traceback (most recent call last)
~\AppData\Local\Temp/ipykernel_34972/2185440898.py in <module>
      1 # Main imports
      2 from econml.dynamic.dml import DynamicDML
----> 3 from econml.tests.dgp import DynamicPanelDGP, add_vlines
      4 
      5 # Helper imports

ModuleNotFoundError: No module named 'econml.tests'

Any help would be appreciated.