update dependency remove click

Author: liugangcode

pyproject.toml

@@ -19,11 +19,9 @@ dependencies = [
     "torch>=2.2.0",
     "torch-geometric>=2.6.1",
     "numpy",
-    "click",
     "huggingface_hub>=0.22.2",
     "joblib>=1.3.2",
     "networkx>=3.2.1",
-    "PyYAML>=6.0.2",
     "rdkit>=2023.9.5",
     "scikit_learn>=1.4.1.post1",
     "scipy>=1.14.1",

tests/local_import.py

@@ -1,8 +1,18 @@
 import os
 import ast
 
-def files_importing_pandas(root_dir):
-    pandas_files = set()
+# List of packages from dependencies in pyproject.toml
+packages = [
+    'click',
+    'joblib',
+    'networkx',
+    'yaml',  # PyYAML
+    'sklearn',
+    'scipy',
+]
+
+def files_importing_package(root_dir, package):
+    package_files = set()
     for dirpath, dirnames, filenames in os.walk(root_dir):
         for fn in filenames:
             if not fn.endswith('.py'):
@@ -14,21 +24,28 @@ def files_importing_pandas(root_dir):
             except (SyntaxError, UnicodeDecodeError):
                 continue
             for node in ast.walk(tree):
-                # catch “import pandas” or “import pandas as pd”
+                # catch "import package" or "import package as alias"
                 if isinstance(node, ast.Import):
                     for alias in node.names:
-                        if alias.name == 'pandas' or alias.name.startswith('pandas.'):
-                            pandas_files.add(path)
+                        if alias.name == package or alias.name.startswith(package + '.'):
+                            package_files.add(path)
                             break
-                # catch “from pandas import …” or “from pandas.core import …”
+                # catch "from package import …" or "from package.module import …"
                 elif isinstance(node, ast.ImportFrom):
-                    if node.module and node.module.startswith('pandas'):
-                        pandas_files.add(path)
+                    if node.module and node.module.startswith(package):
+                        package_files.add(path)
                         break
-    return pandas_files
+    return package_files
 
 if __name__ == '__main__':
     project_root = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))  # or replace with your path
     print(f"Project root: {project_root}")
-    for f in sorted(files_importing_pandas(project_root)):
-        print(f)
+    
+    for package in packages:
+        print(f"\n=== Files importing {package} ===")
+        files = files_importing_package(project_root, package)
+        if files:
+            for f in sorted(files):
+                print(f)
+        else:
+            print(f"No files found importing {package}")