prefab 1.0.3__py3-none-any.whl → 1.0.4__py3-none-any.whl

prefab/__init__.py

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

prefab/device.py

@@ -224,6 +224,7 @@ class Device(BaseModel):
         model: Model,
         model_type: str,
         binarize: bool,
+        gpu: bool = False,
     ) -> "Device":
         try:
             with open(os.path.expanduser("~/.prefab.toml")) as file:
@@ -245,6 +246,7 @@ class Device(BaseModel):
                 "Signup/login and generate a new token.\n"
                 "See https://www.prefabphotonics.com/docs/guides/quickstart."
             ) from None
+
         headers = {
             "Authorization": f"Bearer {access_token}",
             "X-Refresh-Token": refresh_token,
@@ -258,84 +260,99 @@ class Device(BaseModel):
         }
         json_data = json.dumps(predict_data)
 
-        endpoint_url = "https://prefab-photonics--predict-v1.modal.run"
-
-        with requests.post(
-            endpoint_url, data=json_data, headers=headers, stream=True
-        ) as response:
-            response.raise_for_status()
-            event_type = None
-            model_descriptions = {"p": "Prediction", "c": "Correction", "s": "SEMulate"}
-            progress_bar = tqdm(
-                total=100,
-                desc=f"{model_descriptions[model_type]}",
-                unit="%",
-                colour="green",
-                bar_format="{l_bar}{bar:30}{r_bar}{bar:-10b}",
-            )
+        endpoint_url = (
+            "https://prefab-photonics--predict-gpu-v1.modal.run"
+            if gpu
+            else "https://prefab-photonics--predict-v1.modal.run"
+        )
 
-            for line in response.iter_lines():
-                if line:
-                    decoded_line = line.decode("utf-8").strip()
-                    if decoded_line.startswith("event:"):
-                        event_type = decoded_line.split(":")[1].strip()
-                    elif decoded_line.startswith("data:"):
-                        try:
-                            data_content = json.loads(decoded_line.split("data: ")[1])
-                            if event_type == "progress":
-                                progress = round(100 * data_content["progress"])
-                                progress_bar.update(progress - progress_bar.n)
-                            elif event_type == "result":
-                                results = []
-                                for key in sorted(data_content.keys()):
-                                    if key.startswith("result"):
-                                        decoded_image = self._decode_array(
-                                            data_content[key]
-                                        )
-                                        results.append(decoded_image)
+        try:
+            with requests.post(
+                endpoint_url, data=json_data, headers=headers, stream=True
+            ) as response:
+                response.raise_for_status()
+                event_type = None
+                model_descriptions = {
+                    "p": "Prediction",
+                    "c": "Correction",
+                    "s": "SEMulate",
+                }
+                progress_bar = tqdm(
+                    total=100,
+                    desc=f"{model_descriptions[model_type]}",
+                    unit="%",
+                    colour="green",
+                    bar_format="{l_bar}{bar:30}{r_bar}{bar:-10b}",
+                )
 
-                                if results:
-                                    prediction = np.stack(results, axis=-1)
-                                    if binarize:
-                                        prediction = geometry.binarize_hard(prediction)
+                for line in response.iter_lines():
+                    if line:
+                        decoded_line = line.decode("utf-8").strip()
+                        if decoded_line.startswith("event:"):
+                            event_type = decoded_line.split(":")[1].strip()
+                        elif decoded_line.startswith("data:"):
+                            try:
+                                data_content = json.loads(
+                                    decoded_line.split("data: ")[1]
+                                )
+                                if event_type == "progress":
+                                    progress = round(100 * data_content["progress"])
+                                    progress_bar.update(progress - progress_bar.n)
+                                elif event_type == "result":
+                                    results = []
+                                    for key in sorted(data_content.keys()):
+                                        if key.startswith("result"):
+                                            decoded_image = self._decode_array(
+                                                data_content[key]
+                                            )
+                                            results.append(decoded_image)
+
+                                    if results:
+                                        prediction = np.stack(results, axis=-1)
+                                        if binarize:
+                                            prediction = geometry.binarize_hard(
+                                                prediction
+                                            )
+                                        progress_bar.close()
+                                        return prediction
+                                elif event_type == "end":
+                                    print("Stream ended.")
                                     progress_bar.close()
-                                    return prediction
-                            elif event_type == "end":
-                                print("Stream ended.")
-                                progress_bar.close()
-                                break
-                            elif event_type == "auth":
-                                if "new_refresh_token" in data_content["auth"]:
-                                    prefab_file_path = os.path.expanduser(
-                                        "~/.prefab.toml"
-                                    )
-                                    with open(
-                                        prefab_file_path, "w", encoding="utf-8"
-                                    ) as toml_file:
-                                        toml.dump(
-                                            {
-                                                "access_token": data_content["auth"][
-                                                    "new_access_token"
-                                                ],
-                                                "refresh_token": data_content["auth"][
-                                                    "new_refresh_token"
-                                                ],
-                                            },
-                                            toml_file,
+                                    break
+                                elif event_type == "auth":
+                                    if "new_refresh_token" in data_content["auth"]:
+                                        prefab_file_path = os.path.expanduser(
+                                            "~/.prefab.toml"
                                         )
-                            elif event_type == "error":
-                                print(f"Error: {data_content['error']}")
-                                progress_bar.close()
-                        except json.JSONDecodeError:
-                            print(
-                                "Failed to decode JSON:",
-                                decoded_line.split("data: ")[1],
-                            )
+                                        with open(
+                                            prefab_file_path, "w", encoding="utf-8"
+                                        ) as toml_file:
+                                            toml.dump(
+                                                {
+                                                    "access_token": data_content[
+                                                        "auth"
+                                                    ]["new_access_token"],
+                                                    "refresh_token": data_content[
+                                                        "auth"
+                                                    ]["new_refresh_token"],
+                                                },
+                                                toml_file,
+                                            )
+                                elif event_type == "error":
+                                    raise ValueError(f"{data_content['error']}")
+                            except json.JSONDecodeError:
+                                raise ValueError(
+                                    "Failed to decode JSON:",
+                                    decoded_line.split("data: ")[1],
+                                ) from None
+        except requests.RequestException as e:
+            raise RuntimeError(f"Request failed: {e}") from e
 
     def predict(
         self,
         model: Model,
         binarize: bool = False,
+        gpu: bool = False,
     ) -> "Device":
         """
         Predict the nanofabrication outcome of the device using a specified model.
@@ -357,6 +374,10 @@ class Device(BaseModel):
             If True, the predicted device geometry will be binarized using a threshold
             method. This is useful for converting probabilistic predictions into binary
             geometries. Defaults to False.
+        gpu : bool, optional
+            If True, the prediction will be performed on a GPU. Defaults to False.
+            Note: The GPU option has more overhead and will take longer for small
+            devices, but will be faster for larger devices.
 
         Returns
         -------
@@ -373,6 +394,7 @@ class Device(BaseModel):
             model=model,
             model_type="p",
             binarize=binarize,
+            gpu=gpu,
         )
         return self.model_copy(update={"device_array": prediction_array})
 
@@ -380,6 +402,7 @@ class Device(BaseModel):
         self,
         model: Model,
         binarize: bool = True,
+        gpu: bool = False,
     ) -> "Device":
         """
         Correct the nanofabrication outcome of the device using a specified model.
@@ -403,6 +426,10 @@ class Device(BaseModel):
             If True, the corrected device geometry will be binarized using a threshold
             method. This is useful for converting probabilistic corrections into binary
             geometries. Defaults to True.
+        gpu : bool, optional
+            If True, the prediction will be performed on a GPU. Defaults to False.
+            Note: The GPU option has more overhead and will take longer for small
+            devices, but will be faster for larger devices.
 
         Returns
         -------
@@ -419,12 +446,14 @@ class Device(BaseModel):
             model=model,
             model_type="c",
             binarize=binarize,
+            gpu=gpu,
         )
         return self.model_copy(update={"device_array": correction_array})
 
     def semulate(
         self,
         model: Model,
+        gpu: bool = False,
     ) -> "Device":
         """
         Simulate the appearance of the device as if viewed under a scanning electron
@@ -444,6 +473,10 @@ class Device(BaseModel):
             in `models.py`. Each model is associated with a version and dataset that
             detail its creation and the data it was trained on, ensuring the SEMulation
             is tailored to specific fabrication parameters.
+        gpu : bool, optional
+            If True, the prediction will be performed on a GPU. Defaults to False.
+            Note: The GPU option has more overhead and will take longer for small
+            devices, but will be faster for larger devices.
 
         Returns
         -------
@@ -455,6 +488,7 @@ class Device(BaseModel):
             model=model,
             model_type="s",
             binarize=False,
+            gpu=gpu,
         )
         return self.model_copy(update={"device_array": semulated_array})
 
@@ -1259,3 +1293,5 @@ class Device(BaseModel):
             with higher values indicating greater uncertainty.
         """
         return 1 - 2 * np.abs(0.5 - self.device_array)
+
+        return 1 - 2 * np.abs(0.5 - self.device_array)

{prefab-1.0.3.dist-info → prefab-1.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prefab
-Version: 1.0.3
+Version: 1.0.4
 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.0.3.dist-info → prefab-1.0.4.dist-info}/RECORD

@@ -1,12 +1,12 @@
-prefab/__init__.py,sha256=54JuP2jMRBoA7uWb9uoQ_VXXsi_V1ld_LM2A9wXiLkA,401
+prefab/__init__.py,sha256=z7di6Rcufbv_haAVfAam3Q_EZaBWjv-DWpidsqLlBmc,401
 prefab/__main__.py,sha256=aAgt1WXa44k1nJqsiSD3uAfNeGpwtjWqMUYCHN5_Qrw,2759
 prefab/compare.py,sha256=AfLJ69DTqvt0mkMqNCTdn2KWKoclt_0htGVXvarEhkY,2709
-prefab/device.py,sha256=sWDL9xXjDs0LzL91oGu8JFvYCQcMF8msz7fjbxwDWEg,48397
+prefab/device.py,sha256=qIPQNt3nqIpeZasBWHNFxdHC6EBt1_fgWdmeUVWDPlE,50109
 prefab/geometry.py,sha256=pXsVeu4Ycnq60bG6WqFNoUWnyiNStIaYQgbMIrEyndM,9614
 prefab/models.py,sha256=UMzYZzKouroxlwkXCMKIYozmQCMhNhvt8kQrZmwmZB4,3671
 prefab/read.py,sha256=HNuovvQx0Vg0X-z5-gUlL-T1M1RQf81Bmy0wDAdqO28,10782
 prefab/shapes.py,sha256=Hc6dc2Y5Wmb2mZAxhdjBNEJF7C7tF2o460HNvcqcQdo,24856
-prefab-1.0.3.dist-info/METADATA,sha256=yQgJ6qqL8WWfcZmVxRpmAvucNq1XSBYLQra4fmpH3T0,34796
-prefab-1.0.3.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-prefab-1.0.3.dist-info/licenses/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-prefab-1.0.3.dist-info/RECORD,,
+prefab-1.0.4.dist-info/METADATA,sha256=DI5SMrZH42zLQnaJlzzKkU5BU9KEpAJsSayAvIxIAk0,34796
+prefab-1.0.4.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+prefab-1.0.4.dist-info/licenses/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+prefab-1.0.4.dist-info/RECORD,,