fix: swapped _R_co and _P to prevent type error. (Lightning-AI#20508)

When loading a pytorch-lightning model from MLFlow, I get `TypeError: Type parameter +_R_co without a default follows type parameter with a default`. This happens whenever doing `import pytorch_lightning as pl` which is done by packages like MLFlow. Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com> Co-authored-by: Luca Antiga <luca.antiga@gmail.com>
newbienewbie · Jan 6, 2025 · ee7fa43 · ee7fa43
1 parent efe311c
commit ee7fa43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lightning/pytorch/utilities/model_helpers.py b/src/lightning/pytorch/utilities/model_helpers.py
@@ -104,7 +104,7 @@ def _check_mixed_imports(instance: object) -> None:
 _R_co = TypeVar("_R_co", covariant=True)  # return type of the decorated method
 
 
-class _restricted_classmethod_impl(Generic[_T, _P, _R_co]):
+class _restricted_classmethod_impl(Generic[_T, _R_co, _P]):
     """Drop-in replacement for @classmethod, but raises an exception when the decorated method is called on an instance
     instead of a class type."""