alt to rstrip

mindsdb · Oct 7, 2021 · 36ecb97 · 36ecb97
1 parent 1b9888f
commit 36ecb97
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lightwood/api/json_ai.py b/lightwood/api/json_ai.py
@@ -23,6 +23,8 @@
 for import_dir in [os.path.expanduser('~/lightwood_modules'), '/etc/lightwood_modules']:
     if os.path.exists(import_dir) and os.access(import_dir, os.R_OK):
         for file_name in list(os.walk(import_dir))[0][2]:
+            if file_name[-3:] != '.py':
+                continue
             mod_name = file_name[:-3]
             loader = importlib.machinery.SourceFileLoader(mod_name,
                                                           os.path.join(import_dir, file_name))