Add the hpccm directory to the python module search path when using t…

…he wrapper script (#282)
NVIDIA · Jul 10, 2020 · 37319a0 · 37319a0
1 parent 785f9ed
commit 37319a0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/hpccm.sh b/hpccm.sh
@@ -14,4 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# add the path where this script resides to the search path for python modules
+wd=$(readlink -e $(dirname $0))
+export PYTHONPATH=${wd}:$PYTHONPATH
+
 python3 -m hpccm.cli $*