is there any version supporting GPU core? Currently, i used 'pip install econml' in GCP gpu core machine, but i got some issue when trying 'from econml.dml.dml import LinearDML, SparseLinearDML'

[ysong2123]

[ysong2123]

Append related error log:

from econml.dr import ForestDRLearner Traceback (most recent call last): File "", line 1, in File "/home/econml/econml/dr/init.py", line 4, in from ._drlearner import (DRLearner, LinearDRLearner, SparseLinearDRLearner, ForestDRLearner) File "/home/econml/econml/dr/_drlearner.py", line 51, in from ..grf import RegressionForest File "/home/econml/econml/grf/init.py", line 14, in from ._criterion import LinearMomentGRFCriterion, LinearMomentGRFCriterionMSE ModuleNotFoundError: No module named 'econml.grf._criterion' import numpy as np from econml.grf import RegressionForest Traceback (most recent call last): File "", line 1, in File "/home/econml/econml/grf/init.py", line 14, in from ._criterion import LinearMomentGRFCriterion, LinearMomentGRFCriterionMSE ModuleNotFoundError: No module named 'econml.grf._criterion' from econml.grf._base_grftree import GRFTree Traceback (most recent call last): File "", line 1, in File "/home/econml/econml/grf/init.py", line 14, in from ._criterion import LinearMomentGRFCriterion, LinearMomentGRFCriterionMSE ModuleNotFoundError: No module named 'econml.grf._criterion' from econml.tree._tree import Tree Traceback (most recent call last): File "", line 1, in File "/home/econml/econml/tree/init.py", line 4, in from ._criterion import Criterion, RegressionCriterion, MSE ModuleNotFoundError: No module named 'econml.tree._criterion'

[kbattocchi]

We provide pre-built wheels for the most common Linux, Windows, and macOS architectures, as well as a source distribution for the package which should be able to be installed in most other environments. Can you run pip uninstall econml, and then see if there are any warnings or errors when running pip install econml again? Also, could you provide the output of pip list? Thanks.

[ysong2123]

after running 'pip uninstall econml', and then 'pip install econml', i got: Successfully installed econml-0.13.0. And after running 'pip list', i got: Package Version Editable project location

---

absl-py 0.15.0 aiohttp 3.8.1 aiosignal 1.2.0 alembic 1.7.5 ansiwrap 0.8.4 anyio 3.4.0 APScheduler 3.8.1 argon2-cffi 21.3.0 argon2-cffi-bindings 21.2.0 astunparse 1.6.3 async-generator 1.10 async-timeout 4.0.2 attrs 21.4.0 Babel 2.9.1 backcall 0.2.0 backports.functools-lru-cache 1.6.4 bcrypt 3.2.0 beautifulsoup4 4.6.0 black 21.12b0 bleach 4.1.0 blinker 1.4 brotlipy 0.7.0 cachetools 4.2.4 certifi 2021.10.8 certipy 0.1.3 cffi 1.15.0 charset-normalizer 2.0.9 click 8.0.3 cloud-magic-connection 0.2.3.3 cloud-sdk-python 0.0.0 cloudpickle 2.0.0 colorama 0.4.4 conda 4.11.0 conda-package-handling 1.7.3 cryptography 36.0.1 Cython 0.29.26 databricks-cli 0.16.2 debugpy 1.5.1 decorator 5.1.0 deepdiff 5.7.0 defusedxml 0.7.1 docker 5.0.3 dowhy 0.6 econml 0.13.0 entrypoints 0.3 Flask 2.0.2 flask-swagger-ui 3.36.0 flatbuffers 1.12 flit_core 3.6.0 frozenlist 1.2.0 gast 0.4.0 gimellogger 0.1.3 gitdb 4.0.9 GitPython 3.1.26 google-auth 1.35.0 google-auth-oauthlib 0.4.1 google-pasta 0.2.0 gorilla 0.4.0 greenlet 1.1.2 grpcio 1.34.1 gunicorn 20.1.0 h5py 3.1.0 hdfs3 0.3.1 idna 3.1 importlib-metadata 4.10.0 importlib-resources 5.4.0 ipykernel 6.6.1 ipython 7.31.0 ipython-genutils 0.2.0 ipython-sql 0.3.8 ipywidgets 7.6.5 itsdangerous 2.0.1 jedi 0.18.1 jeepney 0.7.1 Jinja2 3.0.3 jira 3.1.1 joblib 1.1.0 json5 0.9.5 jsonschema 4.3.3 jupyter-client 7.1.0 jupyter-contrib-core 0.3.3 jupyter-contrib-nbextensions 0.5.1 jupyter-core 4.9.1 jupyter-highlight-selected-word 0.2.0 jupyter-latex-envs 1.4.6 jupyter-nbextensions-configurator 0.4.1 jupyter-server 1.13.1

jupyter-telemetry 0.1.0 jupyter-tensorboard 0.2.0 jupyterhub 1.1.0 jupyterlab 2.2.4 jupyterlab-server 1.2.0 jupyterlab-widgets 1.0.2 kafka-python 2.0.2 Keras-Preprocessing 1.1.2 keyring 23.5.0 libmambapy 0.19.1 lightgbm 3.3.2 llvmlite 0.38.0 lxml 4.7.1 Mako 1.1.6 mamba 0.19.1 Markdown 3.3.6 MarkupSafe 2.0.1 matplotlib-inline 0.1.3 mistune 0.8.4 mlflow 1.8.1.dev6 mpmath 1.2.1 multidict 5.2.0 mypy-extensions 0.4.3 mysql-connector-python 8.0.16 nacos-sdk-python 0.1.5 nb2kg 0.7.0 nbclassic 0.3.4 nbclient 0.5.9 nbconvert 5.6.1 nbdime 2.1.1 nbformat 5.1.3 ndg-httpsclient 0.5.1 nest-asyncio 1.5.4 networkx 2.7.1 notebook 5.7.9 numba 0.55.1 numpy 1.19.5 oauthlib 3.1.1 opt-einsum 3.3.0 ordered-set 4.0.2 packaging 21.3 pamela 1.0.0 pandas 1.3.5 pandocfilters 1.5.0 papermill 2.3.3 paramiko 2.9.2 parso 0.8.3 pathspec 0.9.0 patsy 0.5.2 pexpect 4.8.0 pickleshare 0.7.5 pip 21.3.1 platformdirs 2.4.1 prettytable 3.0.0 prometheus-client 0.12.0 prometheus-flask-exporter 0.18.7 prompt-toolkit 3.0.24 protobuf 3.19.3 psutil 5.9.0 psycopg2-binary 2.9.3 ptyprocess 0.7.0 pyasn1 0.4.8 pyasn1-modules 0.2.8 pycosat 0.6.3 pycparser 2.21 pycryptodome 3.12.0 pycurl 7.44.1 pydot 1.4.2 Pygments 2.11.1 PyJWT 2.3.0 PyMySQL 1.0.2 PyNaCl 1.5.0 pyOpenSSL 21.0.0 pyparsing 3.0.6 pyrsistent 0.18.0 pysftp 0.2.9 PySocks 1.7.1 python-dateutil 2.8.2 python-json-logger 2.0.1 pytz 2021.3 pytz-deprecation-shim 0.1.0.post0 PyYAML 6.0 pyzmq 22.3.0 qtconsole 5.2.2 QtPy 2.0.0 querystring-parser 1.2.4 requests 2.26.0 requests-oauthlib 1.3.0 requests-toolbelt 0.9.1 rsa 4.8 ruamel.yaml 0.17.19 ruamel.yaml.clib 0.2.6 ruamel-yaml-conda 0.15.80 scikit-learn 1.0.2 scipy 1.8.0 SecretStorage 3.3.1 Send2Trash 1.8.0 setuptools 60.9.3 setuptools-cythonize 1.0.6 shap 0.38.1 simpervisor 0.4 six 1.15.0 slicer 0.0.7 smmap 5.0.0 sniffio 1.2.0 sparse 0.13.0 SQLAlchemy 1.3.13 sqlparse 0.4.2 statsmodels 0.13.2 sympy 1.10 tabulate 0.8.9 tenacity 8.0.1 tensorboard 2.2.0 tensorboard-data-server 0.6.1 tensorboard-plugin-wit 1.8.1 tensorflow 2.5.0 tensorflow-estimator 2.5.0 teradata 15.10.0.20 termcolor 1.1.0 terminado 0.12.1 testpath 0.5.0 textwrap3 0.9.2 threadpoolctl 3.1.0 tomli 1.2.3 tornado 6.1 tqdm 4.62.3 traitlets 5.1.1 typing_extensions 4.0.1 tzdata 2021.5 tzlocal 4.1 urllib3 1.26.7 wcwidth 0.2.5 webencodings 0.5.1 websocket-client 1.2.3 Werkzeug 2.0.2 wheel 0.37.1 widgetsnbextension 3.5.2 wrapt 1.12.1 yarl 1.7.2 zipp 3.6.0

[kbattocchi]

Were there any warnings generated during the installation of econml? Do you still get the same stack trace when importing ForestDRLearner?

[ysong2123]

[ysong2123]

only import econml, it is ok. but 'from econml.dr import ForestDRLearner ' i got error: 'ModuleNotFoundError: No module named 'econml.dr' '