fix setup.py bugs (#37)

Xilinx · Mar 5, 2021 · b5c3b97 · b5c3b97
1 parent bb86ed1
commit b5c3b97
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -111,10 +111,9 @@ def install_vart_pkg(pkg_path, edge):
               'cd {1} && '
               'apt-get install ./*.deb && '
               'cd ../ && '
-              'rm -rf *.tar.gz aarch64 armv7l'.format(pkg_path, edge))
-    os.system('rm -rf /etc/vart.conf')
-    with open('/etc/vart.conf') as f:
-        f.write("firmware: /usr/lib/dpu.xclbin")
+              'rm -rf *.tar.gz aarch64 armv7l && '
+              'sed -i "s/media\/sd-mmcblk0p1/usr\/lib/g" '
+              '/etc/vart.conf'.format(pkg_path, edge))
 
 
 # resolve overlay files by moving the cached copy