fix location check in download-test-data script

yannvgn · Dec 19, 2019 · bcf6097 · bcf6097
1 parent dbf6972
commit bcf6097
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/laserembeddings/__main__.py b/laserembeddings/__main__.py
@@ -106,7 +106,7 @@ def main():
         repository_root = os.path.dirname(
             os.path.dirname(os.path.realpath(__file__)))
 
-        if os.path.basename(repository_root) != 'laserembeddings':
+        if not os.path.isfile(os.path.join(repository_root, 'pyproject.toml')):
             print(
                 f"{CONSOLE_ERROR}  Looks like you're not running laserembeddings from its source code"
             )