Add figure generation code

src/nwb_benchmarks/database/_models.py

@@ -4,6 +4,7 @@
 import pathlib
 import re
 import uuid
+from datetime import datetime
 
 import packaging.version
 import typing_extensions
@@ -13,7 +14,7 @@
 class Result:
     uuid: str
     version: str
-    timestamp: str
+    timestamp: datetime
     commit_hash: str
     environment_id: str
     machine_id: str
@@ -68,7 +69,7 @@ def parse_parameter_case(s):
             Result(
                 uuid=str(uuid.uuid4()),  # TODO: add this to each results file so it is persistent
                 version=database_version,
-                timestamp=timestamp,
+                timestamp=datetime.strptime(timestamp, "%Y-%m-%d-%H-%M-%S"),
                 commit_hash=commit_hash,
                 environment_id=environment_id,
                 machine_id=machine_id,
@@ -94,6 +95,7 @@ def to_dataframe(self) -> "polars.DataFrame":
             "commit_hash": [result.commit_hash for result in self.results],
             "environment_id": [result.environment_id for result in self.results],
             "machine_id": [result.machine_id for result in self.results],
+            "timestamp": [result.timestamp for result in self.results],
             "benchmark_name": [result.benchmark_name for result in self.results],
             "parameter_case_name": [result.parameter_case.get("name") for result in self.results],
             "parameter_case_https_url": [result.parameter_case.get("https_url") for result in self.results],

src/nwb_benchmarks/database/_processing.py

@@ -58,7 +58,10 @@ def concat_dataclasses_to_parquet(
 
 
 def repackage_as_parquet(
-    directory: pathlib.Path, output_directory: pathlib.Path, minimum_version: str = "1.0.0"
+    directory: pathlib.Path,
+    output_directory: pathlib.Path,
+    minimum_results_version: str = "1.0.0",
+    minimum_machines_version: str = "1.0.0",
 ) -> None:
     """Repackage JSON results files as parquet databases for easier querying."""
 
@@ -69,7 +72,7 @@ def repackage_as_parquet(
         dataclass_name="machines",
         dataclass=Machine,
         concat_how="diagonal_relaxed",
-        minimum_version=minimum_version,
+        minimum_version=minimum_machines_version,
     )
 
     # Environments
@@ -79,7 +82,6 @@ def repackage_as_parquet(
         dataclass_name="environments",
         dataclass=Environment,
         concat_how="diagonal",
-        minimum_version=minimum_version,
     )
 
     # Results
@@ -89,5 +91,5 @@ def repackage_as_parquet(
         dataclass_name="results",
         dataclass=Results,
         concat_how="diagonal_relaxed",
-        minimum_version=minimum_version,
+        minimum_version=minimum_results_version,
     )