Code reformatting

Update pyproject

Author: tennlee

packages/bundled_models/lucie/src/lucie/inference.py

@@ -43,9 +43,18 @@
     torch.cuda.set_device(0)
 
 
-def infer(device,
-    model, steps, initial_frame, forcing, initial_forcing_idx,
-    prog_means, prog_stds, diag_means, diag_stds, diff_stds
+def infer(
+    device,
+    model,
+    steps,
+    initial_frame,
+    forcing,
+    initial_forcing_idx,
+    prog_means,
+    prog_stds,
+    diag_means,
+    diag_stds,
+    diff_stds,
 ):
     inf_data = []
     model.eval()
@@ -82,8 +91,8 @@ def load_data_and_predict(
     device,
     regridded_data,
     preprocessed_data,  # standardised data generated by dataset_generator.py
-    model_weights_pth='model.pth',
-    ):
+    model_weights_pth="model.pth",
+):
 
     regridded_data = regridded_data[..., :6]
     true_clim = torch.tensor(np.mean(regridded_data, axis=0)).to(device).permute(2, 0, 1)

pyproject.toml

@@ -33,7 +33,6 @@ dependencies = [
 "pyearthtools-zoo>=0.5.1",
 ]
 
-# FIXME remove redundant dependencies once PyEarthTools is on PyPI
 [project.optional-dependencies]
 utils = [
     "pyearthtools-utils>=0.5.1",
@@ -42,21 +41,12 @@ data = [
     "pyearthtools-data>=0.5.1",
 ]
 pipeline = [
-    "pyearthtools-utils>=0.5.1",
-    "pyearthtools-data>=0.5.1",
     "pyearthtools-pipeline>=0.5.1",
 ]
 training = [
-    "pyearthtools-utils>=0.5.1",
-    "pyearthtools-data>=0.5.1",
-    "pyearthtools-pipeline>=0.5.1",
     "pyearthtools-training>=0.5.1",
 ]
 zoo = [
-    "pyearthtools-utils>=0.5.1",
-    "pyearthtools-data>=0.5.1",
-    "pyearthtools-pipeline>=0.5.1",
-    "pyearthtools-training>=0.5.1",
     "pyearthtools-zoo>=0.5.1",
 ]
 all = [