Remove context tests and imports in optional duckDB reporter

aai-institute · Oct 26, 2024 · 2fbf462 · 2fbf462
1 parent a6b271b
commit 2fbf462
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 112 deletions.
diff --git a/src/nnbench/reporter/duckdb_sql.py b/src/nnbench/reporter/duckdb_sql.py
@@ -5,7 +5,6 @@
 import weakref
 from pathlib import Path
 
-from nnbench.context import Context
 
 try:
     import duckdb
@@ -122,7 +121,7 @@ def read_sql(
         results = rel.fetchall()
 
         benchmarks = [dict(zip(columns, r)) for r in results]
-        context = Context()
+        context = {}
         for bm in benchmarks:
             context.update(bm.pop("context", {}))
 

diff --git a/tests/test_context.py b/tests/test_context.py
diff --git a/tests/test_fileio.py b/tests/test_fileio.py
@@ -4,7 +4,6 @@
 
 import pytest
 
-from nnbench.context import Context
 from nnbench.reporter.file import FileIO
 from nnbench.types import BenchmarkRecord
 
@@ -20,7 +19,7 @@ def test_fileio_writes_no_compression_inline(
     f = FileIO()
 
     rec = BenchmarkRecord(
-        context=Context.make({"a": "b", "s": 1, "b.c": 1.0}),
+        context={"a": "b", "s": 1, "b.c": 1.0},
         benchmarks=[{"name": "foo", "value": 1}, {"name": "bar", "value": 2}],
     )
     file = tmp_path / f"record.{ext}"