Robust "PYTHONPATH" usage.

1aceb017 · Mark Daoust · 1056788e · 1aceb017
Commit 1aceb017 authored Jul 17, 2018 by Mark Daoust
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 4 deletions

samples/core/tutorials/estimators/linear.ipynb samples/core/tutorials/estimators/linear.ipynb +7 -4

No files found.
--- a/samples/core/tutorials/estimators/linear.ipynb
+++ b/samples/core/tutorials/estimators/linear.ipynb
@@ -247,7 +247,10 @@
        "#export PYTHONPATH=${PYTHONPATH}:\"$(pwd)/models\"\n",
        "#running from python you need to set the `os.environ` or the subprocess will not see the directory.\n",
        "\n",
-        "os.environ['PYTHONPATH'] += os.pathsep +  models_path"
+        "if \"PYTHONPATH\" in os.environ:\n",
+        "  os.environ['PYTHONPATH'] += os.pathsep +  models_path\n",
+        "else:\n",
+        "  os.environ['PYTHONPATH'] = models_path"
      ],
      "execution_count": 0,
      "outputs": []