torch-structure 0.1.0__py3-none-any.whl

torch_structure/__init__.py

@@ -0,0 +1,33 @@
+"""
+TorchStructure is a python package for graph-based autodifferentiable structural design
+and engineering.
+"""
+
+from importlib.metadata import version, PackageNotFoundError
+
+from . import data
+from . import formfinding
+from . import generators
+from . import loader
+from . import message_passing
+from . import loss
+from . import plot
+from . import transforms
+from . import utils
+
+try:
+    __version__ = version(__name__)
+except PackageNotFoundError:
+    __version__ = "0.0.0"
+
+__all__ = [
+    "data",
+    "formfinding",
+    "generators",
+    "loader",
+    "message_passing",
+    "loss",
+    "plot",
+    "transforms",
+    "utils",
+]

torch_structure/data/__init__.py

@@ -0,0 +1,12 @@
+import torch
+from packaging.version import parse
+
+from .data import StructData
+from .dataset import Dataset, save, LegacyDataset
+from .utils import requires_metadata
+from .view import NodeView
+
+__all__ = ["StructData", "Dataset", "save", "LegacyDataset", "requires_metadata", "NodeView"]
+
+if parse(torch.__version__) >= parse("2.4.0"):
+    torch.serialization.add_safe_globals([StructData])