Cast version to int

SleipnirGroup · Nov 22, 2024 · 2aa257b · 2aa257b
1 parent 85d9413
commit 2aa257b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/choreolib/py/choreo/__init__.py b/choreolib/py/choreo/__init__.py
@@ -76,7 +76,7 @@ def load_swerve_trajectory_string(trajectory_json_string: str) -> SwerveTrajecto
     """
     data = json.loads(trajectory_json_string)
     name = data["name"]
-    version = data["version"]
+    version = int(data["version"])
     if version != SPEC_VERSION:
         raise ValueError(
             f"{name}.traj: Wrong version {version}. Expected {SPEC_VERSION}"

diff --git a/choreolib/py/choreo/trajectory/__init__.py b/choreolib/py/choreo/trajectory/__init__.py
@@ -138,7 +138,7 @@ def get_chassis_speeds(self) -> ChassisSpeeds:
 
         with open(chor, "r", encoding="utf-8") as project_file:
             data = json.load(project_file)
-        version = data["version"]
+        version = int(data["version"])
         if version != SPEC_VERSION:
             raise ValueError(
                 f".chor project file: Wrong version {version}. Expected {SPEC_VERSION}"