[Feature] Support pip install (#365)

Author: jiahy0825

docs/README_contribute.md

@@ -49,7 +49,7 @@ import graph_net
 model = ...  
 
 # Extract your own model
-model = graph_net.torch.extract(name="model_name", dynamic="True")(model)
+model = graph_net.torch.extract(name="model_name", dynamic=True)(model)
 ```
 
 After running, the extracted graph will be saved to: `$GRAPH_NET_EXTRACT_WORKSPACE/model_name/`.

graph_net/__init__.py

@@ -1 +1,21 @@
+__all__ = ["torch", "paddle"]
 
+from importlib import import_module
+from typing import TYPE_CHECKING, Any, List
+
+
+def __getattr__(name: str) -> Any:
+    if name in __all__:
+        module = import_module(f"{__name__}.{name}")
+        globals()[name] = module
+        return module
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+def __dir__() -> List[str]:
+    return sorted(list(globals().keys()) + __all__)
+
+
+if TYPE_CHECKING:
+    from . import torch as torch  # type: ignore
+    from . import paddle as paddle  # type: ignore

pyproject.toml

@@ -0,0 +1,33 @@
+[build-system]
+requires = ["hatchling>=1.18", "editables>=0.5"]
+build-backend = "hatchling.build"
+
+[project]
+name = "graph-net"
+version = "0.1.0"
+description = "Graph neural network utilities and models."
+readme = "README.md"
+requires-python = ">=3.8"
+license = { file = "LICENSE" }
+authors = [
+  { name = "GraphNet Authors" }
+]
+dependencies = []
+
+[project.optional-dependencies]
+dev = [
+  "pre-commit>=3.0",
+  "astor>=0.8",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["graph_net"]
+
+[tool.hatch.build.targets.sdist]
+include = [
+  "graph_net",
+  "README.md",
+  "LICENSE",
+]
+
+