prefab 1.4.0__py3-none-any.whl → 1.4.1__py3-none-any.whl

prefab/__init__.py

@@ -5,7 +5,7 @@ Usage:
     import prefab as pf
 """
 
-__version__ = "1.4.0"
+__version__ = "1.4.1"
 
 from . import compare, geometry, predict, read, shapes
 from .device import BufferSpec, Device

prefab/models/__init__.py

@@ -0,0 +1,57 @@
+"""
+Fabrication process model definitions and configurations.
+
+This module automatically discovers and loads all model definitions from
+Python files in the models/ directory.
+"""
+
+import importlib
+from pathlib import Path
+
+from .base import Fab, Model
+
+models = {}
+
+
+def _load_models_from_module(module):
+    """
+    Load all Model instances from a module into the models dict.
+
+    If the module defines a __models__ dict, use that for registration.
+    Otherwise, auto-register all Model instances using their variable names.
+
+    Parameters
+    ----------
+    module : module
+        Python module containing Model instances to register.
+    """
+    if hasattr(module, "__models__"):
+        models_dict = getattr(module, "__models__")
+        for name, obj in models_dict.items():
+            if isinstance(obj, Model):
+                models[name] = obj
+    else:
+        for name, obj in vars(module).items():
+            if isinstance(obj, Model):
+                models[name] = obj
+
+
+models_dir = Path(__file__).parent
+
+for model_file in models_dir.glob("*.py"):
+    if model_file.stem in ("__init__", "base"):
+        continue
+
+    module_name = f".{model_file.stem}"
+    try:
+        module = importlib.import_module(module_name, package=__package__)
+        _load_models_from_module(module)
+    except Exception:
+        pass
+
+
+__all__ = [
+    "Fab",
+    "Model",
+    "models",
+]

prefab/{models.py → models/base.py}

@@ -1,11 +1,8 @@
 """
-Fabrication process model definitions and configurations.
+Base model definitions for fabrication processes.
 
-This module defines the data structures for representing nanofabrication processes
-and their associated machine learning models. It includes Pydantic models for
-fabrication specifications (foundry, process) and versioned model configurations
-(dataset, version, release dates). Pre-configured model instances are provided
-for common fabrication processes.
+This module defines the core data structures for representing nanofabrication
+processes and their associated models.
 """
 
 import json
@@ -64,22 +61,3 @@ class Model(BaseModel):
 
     def to_json(self):
         return json.dumps(self.model_dump(), default=str)
-
-
-Generic = Fab(
-    foundry="Generic",
-    process="SOI",
-)
-
-Generic_SOI_ANF1_d0 = Model(
-    fab=Generic,
-    version="ANF1",
-    version_date=date(2025, 11, 7),
-    dataset="d0",
-    dataset_date=date(2025, 11, 7),
-    tag="",
-)
-
-models = dict(
-    Generic_SOI=Generic_SOI_ANF1_d0,
-)

prefab/models/evaluation.py

@@ -0,0 +1,29 @@
+"""
+Base evaluation models.
+
+Pre-configured model instances for common fabrication processes.
+"""
+
+from datetime import date
+
+from .base import Fab, Model
+
+
+Generic = Fab(
+    foundry="Generic",
+    process="SOI",
+)
+
+Generic_SOI_ANF1_d0 = Model(
+    fab=Generic,
+    version="ANF1",
+    version_date=date(2025, 11, 7),
+    dataset="d0",
+    dataset_date=date(2025, 11, 7),
+    tag="",
+)
+
+# Export models with user-facing names
+__models__ = {
+    "Generic_SOI": Generic_SOI_ANF1_d0,
+}

{prefab-1.4.0.dist-info → prefab-1.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefab
-Version: 1.4.0
+Version: 1.4.1
 Summary: Artificial nanofabrication of integrated photonic circuits using deep learning
 Project-URL: Homepage, https://prefabphotonics.com
 Project-URL: Repository, https://github.com/PreFab-Photonics/PreFab

{prefab-1.4.0.dist-info → prefab-1.4.1.dist-info}/RECORD

@@ -1,15 +1,17 @@
-prefab/__init__.py,sha256=z_ADiFh2q1068jMziubytwv4G1_Wh8wYa3hQCm61Mh8,425
+prefab/__init__.py,sha256=f9FN73XE4ufrKEDyy5rGQW2sWWH-8Y4C_7FYwO946Rs,425
 prefab/__main__.py,sha256=M5ulRfgSMdPjeHPoMGQ3P86zFMJDII5I18NGrpFUcsM,3458
 prefab/compare.py,sha256=aX7nr9tznSebYeeztvqIPz57npnJ4-iUeKEedrZdksE,3676
 prefab/device.py,sha256=1O6vTOq4wQRGVYvFWLH0uj1XhhYCfnDnIapDEYnBKHw,47996
 prefab/geometry.py,sha256=-nTaGjdw3KN1SVoyvqdcrE2GJP7OqPF6ivUhrO78rUk,11244
-prefab/models.py,sha256=YORoAPYC0r60yDShmrZdWOEh9jEBcBMizx4aj8ik27U,2047
 prefab/predict.py,sha256=066X_SnSIHPqaPMbk0Yn_yX_KBfUc9yxrsmP_k56FU0,19015
 prefab/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefab/read.py,sha256=5BmvFemobA72urxs4j2VZRVvanZZGu1mDB1Uh-epyvI,8635
 prefab/shapes.py,sha256=mRGwsPS-A9XsW3jgvUuMCEeNv9BLXsEnJkytlIHUAKE,28802
-prefab-1.4.0.dist-info/METADATA,sha256=1pJWOHiZZ58App0zlSB2rK-3pzV07IO51baZ9ryl-O0,33880
-prefab-1.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-prefab-1.4.0.dist-info/entry_points.txt,sha256=h1_A9O9F3NAIoKXD1RPb3Eo-WCSiHhMB_AnagBi6XTQ,48
-prefab-1.4.0.dist-info/licenses/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-prefab-1.4.0.dist-info/RECORD,,
+prefab/models/__init__.py,sha256=rRrjcOcHPcob98Coksc0tbvYcXbm6SoLEb-Md233Jvo,1391
+prefab/models/base.py,sha256=t4VNMsOztPedj3kN5fZ1-4tk0SRHWrMuqnIVHztsCs4,1514
+prefab/models/evaluation.py,sha256=2_Klui6tY8xPvOSVD8VpZCVAnT1RX15FONqWG-_x-J8,484
+prefab-1.4.1.dist-info/METADATA,sha256=miTQTlHxSgZn57wzltpGpAZ4PgUBW-hVSWmxmSDGqCA,33880
+prefab-1.4.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+prefab-1.4.1.dist-info/entry_points.txt,sha256=h1_A9O9F3NAIoKXD1RPb3Eo-WCSiHhMB_AnagBi6XTQ,48
+prefab-1.4.1.dist-info/licenses/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+prefab-1.4.1.dist-info/RECORD,,

{prefab-1.4.0.dist-info → prefab-1.4.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.27.0
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any