Skip to content

Commit 7ddaeea

Browse files
committed
Update numpy version in pandas benchmarks
1 parent 190e4fc commit 7ddaeea

File tree

1 file changed

+12
-23
lines changed

1 file changed

+12
-23
lines changed

mx.graalpython/mx_graalpython_python_benchmarks.py

Lines changed: 12 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -764,22 +764,17 @@ class PandasSuite(PySuite):
764764
VERSION = "1.5.2"
765765
VERSION_TAG = "v" + VERSION
766766

767-
PREREQUISITES = """
768-
setuptools==63.1.0
769-
wheel==0.37.1
770-
"""
771-
772-
BENCHMARK_REQ = f"""
773-
asv==0.5.1
774-
distlib==0.3.6
775-
filelock==3.8.0
776-
platformdirs==2.5.2
777-
six==1.16.0
778-
virtualenv==20.16.3
779-
jinja2
780-
numpy==1.23.5
781-
pandas=={VERSION}
782-
"""
767+
BENCHMARK_REQ = [
768+
"asv==0.5.1",
769+
"distlib==0.3.6",
770+
"filelock==3.8.0",
771+
"platformdirs==2.5.2",
772+
"six==1.16.0",
773+
"virtualenv==20.16.3",
774+
"jinja2",
775+
f"numpy=={NumPySuite.VERSION}",
776+
f"pandas=={VERSION}",
777+
]
783778

784779
def name(self):
785780
return "pandas-suite"
@@ -859,13 +854,7 @@ def _vmRun(self, vm, workdir, command, benchmarks, bmSuiteArgs):
859854

860855
vm.run(workdir, ["-m", "venv", join(workdir, vm_venv)])
861856
pip = join(workdir, vm_venv, "bin", "pip")
862-
requirements_txt = join(workdir, "requirements.txt")
863-
with open(requirements_txt, "w") as f:
864-
f.write(self.PREREQUISITES)
865-
mx.run([pip, "install", "-r", requirements_txt], cwd=workdir)
866-
with open(requirements_txt, "w") as f:
867-
f.write(self.BENCHMARK_REQ)
868-
mx.run([pip, "install", "-r", requirements_txt], cwd=workdir)
857+
mx.run([pip, "install", *self.BENCHMARK_REQ], cwd=workdir)
869858
mx.run(
870859
[join(workdir, vm_venv, "bin", "asv"), "machine", "--yes"], cwd=benchdir
871860
)

0 commit comments

Comments
 (0)