chore: update uv files for pypi publish and adopt lazy_import in code logics

Author: Concode0

datalib/symbolic_regression.py

@@ -31,8 +31,6 @@
 import torch
 from torch.utils.data import Dataset, DataLoader
 
-import pmlb
-
 logger = logging.getLogger(__name__)
 
 # PMLB GitHub raw URL for datasets not yet in the PyPI index
@@ -116,8 +114,12 @@ def _fetch_pmlb_data(dataset_name: str, cache_dir: str) -> pd.DataFrame:
             return pd.read_csv(cached_path, sep='\t', compression='gzip')
 
     # Try pmlb.fetch_data (works for datasets in the PyPI index)
-    if dataset_name in pmlb.dataset_names:
-        return pmlb.fetch_data(dataset_name, local_cache_dir=cache_dir)
+    try:
+        import pmlb
+        if dataset_name in pmlb.dataset_names:
+            return pmlb.fetch_data(dataset_name, local_cache_dir=cache_dir)
+    except ImportError:
+        pass
 
     # Fallback: fetch directly from PMLB GitHub
     url = f"{_PMLB_GITHUB_RAW}/{dataset_name}/{dataset_name}.tsv.gz"

models/md17/forcenet.py

@@ -18,7 +18,10 @@
 from layers import CliffordLayerNorm
 from layers import BladeSelector
 from functional.activation import GeometricGELU, GeometricSquare
-from torch_geometric.nn import global_add_pool
+try:
+    from torch_geometric.nn import global_add_pool
+except ImportError:
+    global_add_pool = None
 
 
 class GaussianRBF(nn.Module):

pyproject.toml

@@ -4,13 +4,42 @@ version = "1.0.0"
 description = "A PyTorch framework for Geometric Algebra Deep Learning"
 readme = "README.md"
 requires-python = ">=3.9"
+license = { file = "LICENSE" }
+authors = [
+    { name = "Eunkyum Kim", email = "nemonanconcode@gmail.com" },
+]
+keywords = [
+    "geometric-algebra",
+    "clifford-algebra",
+    "deep-learning",
+    "pytorch",
+    "rotor",
+    "multivector",
+    "equivariant",
+]
+classifiers = [
+    "Development Status :: 5 - Production/Stable",
+    "Intended Audience :: Science/Research",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Topic :: Scientific/Engineering :: Mathematics",
+    "License :: OSI Approved :: Apache Software License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+]
 dependencies = [
     "torch>=2.0.0",
     "numpy>=1.26.0",
     "hydra-core>=1.3.2",
     "tqdm>=4.67.3",
 ]
 
+[project.urls]
+Homepage = "https://github.com/Concode0/Versor"
+Repository = "https://github.com/Concode0/Versor"
+
 [project.optional-dependencies]
 sr = [
     "pmlb>=1.0.1.post3",

tasks/__init__.py

@@ -5,10 +5,6 @@
 """
 
 from .base import BaseTask
-from .symbolic_regression import SRTask
-from .md17 import MD17Task
-from .lqa import LQATask
-from .deap_eeg import DEAPEEGTask
 
 __all__ = [
     "BaseTask",
@@ -17,3 +13,19 @@
     "LQATask",
     "DEAPEEGTask",
 ]
+
+
+def __getattr__(name):
+    if name == "SRTask":
+        from .symbolic_regression import SRTask
+        return SRTask
+    if name == "MD17Task":
+        from .md17 import MD17Task
+        return MD17Task
+    if name == "LQATask":
+        from .lqa import LQATask
+        return LQATask
+    if name == "DEAPEEGTask":
+        from .deap_eeg import DEAPEEGTask
+        return DEAPEEGTask
+    raise AttributeError(f"module 'tasks' has no attribute {name!r}")