Prepare for new release

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "simple-shapes-dataset"
-version = "0.2.1"
+version = "0.3.0"
 description = "The Simple Shapes Dataset used in the Global Workspace Paper"
 authors = ["Benjamin Devillers <[email protected]>"]
 license = "MIT"
@@ -15,13 +15,13 @@ transformers = "^4.30.2"
 opencv-python = "^4.7.0.72"
 attributes-to-language = {git = "https://github.com/bdvllrs/attributes-to-language.git", rev = "v0.3.0"}
 auto-sbatch = {git = "https://github.com/bdvllrs/auto-sbatch.git", rev = "v0.4.1"}
-migrate-ckpt = {git = "https://github.com/bdvllrs/migrate-ckpt.git", rev = "main"}
+migrate-ckpt = {git = "https://github.com/bdvllrs/migrate-ckpt.git", rev = "v0.2.0"}
 cfg-tools = {git = "https://github.com/bdvllrs/cfg-tools.git", rev = "v0.1.1"}
 torchdata = "^0.6.1"
 pillow = "^9.5.0"
 numpy = "^1.25"
 torch = "^2.0.1"
-shimmer = {git = "[email protected]:bdvllrs/shimmer.git", rev = "main"}
+shimmer = {git = "[email protected]:bdvllrs/shimmer.git", rev = "v0.5.1"}
 wandb = "^0.15.4"
 lightning = "^2.1.0"
 pydantic = "^2.6.0"