asim 25.12.1__cp313-cp313-manylinux2014_x86_64.manylinux_2_17_x86_64.whl

asim/__init__.py

@@ -0,0 +1 @@
+from .version import __version__

asim/dataset.pyi

@@ -0,0 +1,8 @@
+from typing import List
+
+import pandas as pd
+
+class PhysicalDataManage:
+    def __init__(self, data: pd.DataFrame, columns: List, batch_size: int = 64): ...
+    @property
+    def df(self) -> pd.DataFrame: ...

asim/model.pyi

@@ -0,0 +1,9 @@
+from io import BytesIO
+from typing import Union
+
+from .dataset import PhysicalDataManage
+
+class PhysicalFieldModel:
+    def __init__(self, dm: PhysicalDataManage, lr: float): ...
+    def fit(self, epochs: int): ...
+    def export(self, fh: Union[str, BytesIO]): ...

asim/simulator.pyi

@@ -0,0 +1,37 @@
+from io import BytesIO
+from typing import Dict, Union, List
+
+import numpy as np
+import onnxruntime as ort
+import pandas as pd
+
+from .dataset import PhysicalDataManage
+
+class PhysicalSimulator:
+
+    _sess: ort.InferenceSession
+    _dm: PhysicalDataManage
+
+    def __init__(self, fh: Union[str, BytesIO], dm: PhysicalDataManage):
+        """
+        init
+        """
+        ...
+
+    def get_group_names(self) -> List[List[str]]:
+        """
+        get_group_names
+        """
+        ...
+
+    def step(self, x1: Dict[str, np.ndarray], y0: Dict[str, np.ndarray]) -> Dict[str, list]:
+        """
+        sim.step({"load": 150.0}, {"power1": 60.0, "power2": 80.0})
+        """
+        ...
+
+    def steps(self, df: pd.DataFrame, x0: Dict[str, float]) -> pd.DataFrame:
+        """
+        steps
+        """
+        ...

asim/version.py

@@ -0,0 +1 @@
+__version__ = "25.12.1"

asim-25.12.1.dist-info/METADATA

@@ -0,0 +1,78 @@
+Metadata-Version: 2.4
+Name: asim
+Version: 25.12.1
+Summary: Automatic simulation system powered by neural networks
+Keywords: simulation
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch>=2.8.0
+Requires-Dist: lightning>=2.5.5
+Provides-Extra: dev
+Requires-Dist: matplotlib; extra == "dev"
+Dynamic: license-file
+
+# Automatic simulation system powered by neural networks
+
+## Installation
+
+```bash
+pip install asim
+```
+
+## What is asim
+
+- Physical field modeling with automatic constraint enforcement
+- Flexible data grouping and normalization
+- Built-in support for recurrent architectures
+- Export/import of trained models
+
+## Using asim
+
+```python
+import matplotlib.pyplot as plt
+import pandas as pd
+from asim.dataset import PhysicalDataManage, Ts, Fi, Fo, DummyDatasets
+from asim.model import PhysicalFieldModel
+from asim.optimizer import ContinuousOptimizer
+from asim.simulator import PhysicalSimulator
+
+# 1. Define data and structure
+df = DummyDatasets.basic_boiler(size=1000)  # pd.read_csv("demo.csv")
+cols = [
+    Ts(label="ts"),
+    Fi(group="boiler", label="power1", min=0.0, union="kw", control=True),
+    Fi(group="boiler", label="power2", min=0.0, union="kw", control=True),
+    Fo(group="boiler", label="load", min=0.0, union="t", loop=True),
+]
+dm = PhysicalDataManage(df, columns=cols, batch_size=64)
+
+# 2A. Select the model, define the parameters, train and save
+fm = PhysicalFieldModel(dm, lr=0.003)
+fm.fit(epochs=100)
+fm.export("demo.sim.onnx")
+
+# 2B. Use a simulator to simulate the operation
+sim = PhysicalSimulator("demo.sim.onnx", dm=dm)
+sim_df = df[100:310].copy()
+sim_df["ts"] = pd.to_datetime(sim_df["ts"], unit="s")
+sim_df = sim.steps(sim_df, x0={"load": 150.0})
+sim_df.plot(
+    x="ts",
+    y=sum(sim.get_group_names(), []),
+    subplots=sim.get_group_names(),
+    sharex=True,
+    figsize=(10, 3),
+    ylim=(0, None),
+    grid=True,
+)
+plt.tight_layout()
+plt.show()
+
+# 3A. RL
+opt = ContinuousOptimizer(sim=sim, dm=dm)
+opt.fit(epochs=100)
+opt.export("demo.opt.onnx")
+
+# 3B. Use the simulator to reinforce the test
+```

asim-25.12.1.dist-info/RECORD

@@ -0,0 +1,15 @@
+asim/__init__.py,sha256=Jzb_RfgvXCrm_SQ4AfeGVi1N36YybxnM5mpyxrnihgI,33
+asim/dataset.cpython-313-x86_64-linux-gnu.so,sha256=gqiFN_sm5X7UHBYL7WW-0frDzh012F2-pPKQQaM_GAw,1372528
+asim/dataset.pyi,sha256=NZnlI9dEfYCP-kA8-BqjzFREClkwkEDaLQ_n0T78L9s,209
+asim/explainer.cpython-313-x86_64-linux-gnu.so,sha256=MqeCPJ9UuZBQoa9GAf-NYwCn3rAbGllyR17hRx0KPuw,193456
+asim/model.cpython-313-x86_64-linux-gnu.so,sha256=9Lmgp4MXi5RNE-XrPrTAOx_WM6OV0GEr9esvRk8NdTc,1392712
+asim/model.pyi,sha256=qe0u6xUiv2yzFJWnScowi0_6LE6MAO2C8I4Rh8gKxBg,266
+asim/optimizer.cpython-313-x86_64-linux-gnu.so,sha256=fDeqloPXjd3cHR5wDq6gO13SiuZy55nak0rCPBiF1Qc,235352
+asim/simulator.cpython-313-x86_64-linux-gnu.so,sha256=KE-T8mRyDF9415ipUH83Ao2BNsYI6HIK_fmh1-37EJM,635144
+asim/simulator.pyi,sha256=MrpJgkKo9DA5hi7QPHUr8zyPf8PaRtbts1uHmb8_0Ak,814
+asim/version.py,sha256=rW4VI6vW4cu-DMZjtYqIe3FzMgXgKWI9l9RpcuMnVfw,24
+asim-25.12.1.dist-info/METADATA,sha256=y-kDqVm4SexBRu0omFKTmTResmybUzWb-tDCWJfJ7Sc,2096
+asim-25.12.1.dist-info/WHEEL,sha256=sLnxkimgGZnkHfiVG0CKXk3cbEzmkkBUHrkKJEO9o6g,151
+asim-25.12.1.dist-info/top_level.txt,sha256=6tOoh20rJ_xz66lNdZVG11p8YArVbqac0k2MEp33X9M,5
+asim-25.12.1.dist-info/RECORD,,
+asim-25.12.1.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0

asim-25.12.1.dist-info/WHEEL

@@ -0,0 +1,6 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: false
+Tag: cp313-cp313-manylinux_2_17_x86_64
+Tag: cp313-cp313-manylinux2014_x86_64
+

asim-25.12.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+asim