Fix non-editable install

Signed-off-by: Keith Battocchi <kebatt@microsoft.com>

Author: kbattocchi

.github/workflows/ci.yml

@@ -125,7 +125,7 @@ jobs:
         - kind: "customer-scenarios"
           extras: "[plt,dowhy]"
           pattern: "CustomerScenarios"
-          version: '3.9'
+          version: '3.11'
           install_graphviz: false
       fail-fast: false
     env:
@@ -145,9 +145,13 @@ jobs:
     - name: Install graphviz
       run: sudo apt-get -yq install graphviz
       if: ${{ matrix.install_graphviz }}
-    # Add verbose flag to pip installation if in debug mode
     - name: Install econml
-      run: uv pip install --system .${{ matrix.extras }} ${{ fromJSON('["","-v"]')[runner.debug] }} ${{ env.use_lkg && '-r lkg-notebook.txt' }}
+      run: |
+        uv pip install --system build
+        python -m build --installer uv .
+        package=$(ls dist/*.whl)
+        echo "package=$package"
+        uv pip install --system ./$package${{ matrix.extras }} ${{ fromJSON('["","-v"]')[runner.debug] }} ${{ env.use_lkg && '-r lkg-notebook.txt' }}
     # Install notebook requirements (if not already done as part of lkg)
     - name: Install notebook requirements
       run: uv pip install --system jupyter jupyter-client nbconvert nbformat seaborn xgboost tqdm
@@ -162,7 +166,7 @@ jobs:
     - name: Install pytest
       run: uv pip install --system pytest pytest-xdist pytest-cov coverage[toml]
     - name: Run notebook tests
-      run: python -m pytest
+      run: python -P -m pytest
       id: run_tests
       env:
         PYTEST_ADDOPTS: '-m "notebook"'

econml/tests/dgp.py

@@ -176,44 +176,3 @@ def policy_gen(Tpre, X, period):
         return self._gen_data_with_policy(n_units, policy_gen, random_seed=random_seed)
 
 
-# Auxiliary function for adding xticks and vertical lines when plotting results
-# for dynamic dml vs ground truth parameters.
-def add_vlines(n_periods, n_treatments, hetero_inds):
-    locs, labels = plt.xticks([], [])
-    locs += [- .501 + (len(hetero_inds) + 1) / 2]
-    labels += ["\n\n$\\tau_{{{}}}$".format(0)]
-    locs += [qx for qx in np.arange(len(hetero_inds) + 1)]
-    labels += ["$1$"] + ["$x_{{{}}}$".format(qx) for qx in hetero_inds]
-    for q in np.arange(1, n_treatments):
-        plt.axvline(x=q * (len(hetero_inds) + 1) - .5,
-                    linestyle='--', color='red', alpha=.2)
-        locs += [q * (len(hetero_inds) + 1) - .501 + (len(hetero_inds) + 1) / 2]
-        labels += ["\n\n$\\tau_{{{}}}$".format(q)]
-        locs += [(q * (len(hetero_inds) + 1) + qx)
-                 for qx in np.arange(len(hetero_inds) + 1)]
-        labels += ["$1$"] + ["$x_{{{}}}$".format(qx) for qx in hetero_inds]
-    locs += [- .501 + (len(hetero_inds) + 1) * n_treatments / 2]
-    labels += ["\n\n\n\n$\\theta_{{{}}}$".format(0)]
-    for t in np.arange(1, n_periods):
-        plt.axvline(x=t * (len(hetero_inds) + 1) *
-                    n_treatments - .5, linestyle='-', alpha=.6)
-        locs += [t * (len(hetero_inds) + 1) * n_treatments - .501 +
-                 (len(hetero_inds) + 1) * n_treatments / 2]
-        labels += ["\n\n\n\n$\\theta_{{{}}}$".format(t)]
-        locs += [t * (len(hetero_inds) + 1) *
-                 n_treatments - .501 + (len(hetero_inds) + 1) / 2]
-        labels += ["\n\n$\\tau_{{{}}}$".format(0)]
-        locs += [t * (len(hetero_inds) + 1) * n_treatments +
-                 qx for qx in np.arange(len(hetero_inds) + 1)]
-        labels += ["$1$"] + ["$x_{{{}}}$".format(qx) for qx in hetero_inds]
-        for q in np.arange(1, n_treatments):
-            plt.axvline(x=t * (len(hetero_inds) + 1) * n_treatments + q * (len(hetero_inds) + 1) - .5,
-                        linestyle='--', color='red', alpha=.2)
-            locs += [t * (len(hetero_inds) + 1) * n_treatments + q *
-                     (len(hetero_inds) + 1) - .501 + (len(hetero_inds) + 1) / 2]
-            labels += ["\n\n$\\tau_{{{}}}$".format(q)]
-            locs += [t * (len(hetero_inds) + 1) * n_treatments + (q * (len(hetero_inds) + 1) + qx)
-                     for qx in np.arange(len(hetero_inds) + 1)]
-            labels += ["$1$"] + ["$x_{{{}}}$".format(qx) for qx in hetero_inds]
-    plt.xticks(locs, labels)
-    plt.tight_layout()