Merge branch 'main' into batch_transfer_update

openclimatefix · Dec 21, 2023 · de08431 · de08431
2 parents 63f1711 + 0de6f9f
commit de08431
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/.bumpversion.cfg b/.bumpversion.cfg
@@ -1,7 +1,7 @@
 [bumpversion]
 commit = True
 tag = True
-current_version = 2.6.2
+current_version = 2.6.3
 message = Bump version: {current_version} → {new_version} [skip ci]
 
 [bumpversion:file:pvnet/__init__.py]

diff --git a/pvnet/__init__.py b/pvnet/__init__.py
@@ -1,2 +1,2 @@
 """PVNet"""
-__version__ = "2.6.2"
+__version__ = "2.6.3"
diff --git a/tests/test_end2end.py b/tests/test_end2end.py
@@ -0,0 +1,6 @@
+import lightning
+
+
+def test_model_trainer_fit(multimodal_model, sample_train_val_datamodule):
+    trainer = lightning.pytorch.trainer.trainer.Trainer(fast_dev_run=True)
+    trainer.fit(model=multimodal_model, datamodule=sample_train_val_datamodule)