From 94a93870bc1d75a854647636a9bd92788e6e8633 Mon Sep 17 00:00:00 2001
From: Brent Yi <yibrenth@gmail.com>
Date: Sat, 22 May 2021 00:49:43 -0700
Subject: [PATCH] Add missing ClassVar hints

---
 jaxlie/_base.py | 10 +++++-----
 setup.py        |  2 +-
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/jaxlie/_base.py b/jaxlie/_base.py
index 3332898..6d46b7f 100644
--- a/jaxlie/_base.py
+++ b/jaxlie/_base.py
@@ -1,5 +1,5 @@
 import abc
-from typing import Generic, Type, TypeVar, overload
+from typing import ClassVar, Generic, Type, TypeVar, overload
 
 import jax
 import numpy as onp
@@ -18,16 +18,16 @@ class MatrixLieGroup(abc.ABC, EnforceOverrides):
     # Class properties
     # > These will be set in `_utils.register_lie_group()`
 
-    matrix_dim: int
+    matrix_dim: ClassVar[int]
     """Dimension of square matrix output from `.as_matrix()`."""
 
-    parameters_dim: int
+    parameters_dim: ClassVar[int]
     """Dimension of underlying parameters, `.parameters()`."""
 
-    tangent_dim: int
+    tangent_dim: ClassVar[int]
     """Dimension of tangent space."""
 
-    space_dim: int
+    space_dim: ClassVar[int]
     """Dimension of coordinates that can be transformed."""
 
     def __init__(self, parameters: hints.Vector):
diff --git a/setup.py b/setup.py
index acd07a8..0fafe37 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name="jaxlie",
-    version="1.2.3",
+    version="1.2.4",
     description="Matrix Lie groups in Jax",
     long_description=long_description,
     long_description_content_type="text/markdown",