pytorch DeepIV WIP

Signed-off-by: Keith Battocchi <[email protected]>

Author: kbattocchi

.github/workflows/ci.yml

@@ -118,7 +118,7 @@ jobs:
         kind: [except-customer-scenarios, customer-scenarios]
         include:
         - kind: "except-customer-scenarios"
-          extras: "[tf,plt,ray]"
+          extras: "[nn,plt,ray]"
           pattern: "(?!CustomerScenarios)"
           install_graphviz: true
           version: '3.8' # no supported version of tensorflow for 3.9
@@ -226,16 +226,16 @@ jobs:
             extras: ""
           - kind: other
             opts: '-m "cate_api and not ray" -n auto'
-            extras: "[tf,plt]"
+            extras: "[nn,plt]"
           - kind: dml
             opts: '-m "dml and not ray"'
-            extras: "[tf,plt]"
+            extras: "[nn,plt]"
           - kind: main
             opts: '-m "not (notebook or automl or dml or serial or cate_api or treatment_featurization or ray)" -n 2'
-            extras: "[tf,plt,dowhy]"
+            extras: "[nn,plt,dowhy]"
           - kind: treatment
             opts: '-m "treatment_featurization and not ray" -n auto'
-            extras: "[tf,plt]"
+            extras: "[nn,plt]"
           - kind: ray
             opts: '-m "ray"'
             extras: "[ray]"

econml/iv/nnet/__init__.py

@@ -1,6 +1,6 @@
 # Copyright (c) PyWhy contributors. All rights reserved.
 # Licensed under the MIT License.
 
-from ._deepiv import DeepIV
+from ._deepiv import DeepIV, MixtureOfGaussiansModule
 
-__all__ = ["DeepIV"]
+__all__ = ["DeepIV, MixtureOfGaussiansModule"]