install requires
3 changed files+20−20
benchcompress/pyproject.tomlmodified+2−1View file
@@ -21,7 +21,8 @@ dependencies = [
2121 "brotli",
2222 "click",
2323 "numba",
24- "pybind11>=2.11.1"
24+ "pybind11>=2.11.1",
25+ "segyio"
2526 ]
2627
2728 [tool.scikit-build]
benchcompress/setup.pymodified+0−12View file
@@ -32,18 +32,6 @@ setup(
3232 version="0.1.0",
3333 packages=find_packages(where="src"),
3434 package_dir={"": "src"},
35- install_requires=[
36- "numpy",
37- "scipy",
38- "zstandard",
39- "simple_ans",
40- "requests",
41- "lindi",
42- "brotli",
43- "click",
44- "pybind11>=2.11.1",
45- "segyio"
46- ],
4735 ext_modules=[
4836 CMakeExtension("benchcompress.algorithms.ans.markov_reconstruct_cpp_ext",
4937 sourcedir="src/benchcompress/algorithms/ans"),
benchcompress/src/benchcompress/run_benchmarks.pymodified+18−7View file
@@ -30,13 +30,24 @@ def is_compatible(algorithm_tags: List[str], dataset_tags: List[str]) -> bool:
3030 Returns:
3131 True if the algorithm should be applied to the dataset
3232 """
33- # If algorithm has delta_encoding tag, dataset must have continuous tag
34- if "delta_encoding" in algorithm_tags and "continuous" not in dataset_tags:
35- return False
36- if "markov_prediction" in algorithm_tags and "continuous" not in dataset_tags:
37- return False
38- if "zero_rle" in algorithm_tags and "sparse" not in dataset_tags:
39- return False
33+ # If algorithm has delta_encoding or markov_prediction, dataset must have continuous, timeseries, 1d
34+ if "delta_encoding" in algorithm_tags or "markov_prediction" in algorithm_tags:
35+ if (
36+ "continuous" not in dataset_tags
37+ or "timeseries" not in dataset_tags
38+ or "1d" not in dataset_tags
39+ ):
40+ return False
41+
42+ # If algorithm has zero_rle, dataset must have sparse, timeseries, 1d
43+ if "zero_rle" in algorithm_tags:
44+ if (
45+ "sparse" not in dataset_tags
46+ or "timeseries" not in dataset_tags
47+ or "1d" not in dataset_tags
48+ ):
49+ return False
50+
4051 return True
4152
4253