Make compatible with older PyTorch versions (#443)

ktschuett · web-flow · commit a7c6bf05d44e · 2022-10-06T10:56:12.000+02:00
* Add workaround for pytorch issue with serializing dtype

* Add some dtype aliases

* Add all dtypes

* Make compatible with earlier torch versions
diff --git a/src/schnetpack/utils/__init__.py b/src/schnetpack/utils/__init__.py
@@ -27,8 +27,6 @@
     "qint8": torch.qint8,
     "qint32": torch.qint32,
     "bool": torch.bool,
-    "quint4x2": torch.quint4x2,
-    "quint2x4": torch.quint2x4,
 }
 
 TORCH_DTYPES.update({"torch." + k: v for k, v in TORCH_DTYPES.items()})

Original file line number	Diff line number	Diff line change
`@@ -27,8 +27,6 @@`
`27`	`27`	`"qint8": torch.qint8,`
`28`	`28`	`"qint32": torch.qint32,`
`29`	`29`	`"bool": torch.bool,`
`30`		`- "quint4x2": torch.quint4x2,`
`31`		`- "quint2x4": torch.quint2x4,`
`32`	`30`	`}`
`33`	`31`
`34`	`32`	`TORCH_DTYPES.update({"torch." + k: v for k, v in TORCH_DTYPES.items()})`