0.5.1 fixes

Author: Zymrael

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "torchdyn"
-version = "0.5.0"
+version = "0.5.1"
 license = "Apache License, Version 2.0"
 description = "Your PyTorch package for neural differential equations"
 authors = ["Michael Poli", "Stefano Massaroli", ]

setup.py

@@ -15,7 +15,7 @@
 
 setup(
     name="torchdyn",
-    version="0.5.0",
+    version="0.5.1",
     author="Michael Poli and Stefano Massaroli",
     description="PyTorch package for all things neural differential equations.",
     url="https://github.com/DiffEqML/torchdyn",