Skip to content

Commit

Permalink
TEST-#2670: use env variables for default values
Browse files Browse the repository at this point in the history
Signed-off-by: Anatoly Myachev <[email protected]>
  • Loading branch information
anmyachev committed Feb 3, 2021
1 parent b276b0f commit 50f76c5
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions asv_bench/benchmarks/benchmarks.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
# define `MODIN_ASV_USE_IMPL` env var to choose library for using in performance
# measurements

import os
import modin.pandas as pd
import numpy as np
import pandas
Expand Down Expand Up @@ -45,8 +46,8 @@
except ImportError:
# The same benchmarking code can be run for different versions of Modin, so in
# case of an error importing important variables, we'll just use predefined values
ASV_USE_IMPL = "modin"
ASV_DATASET_SIZE = "Small"
ASV_USE_IMPL = os.environ.get("MODIN_ASV_USE_IMPL", "modin")
ASV_DATASET_SIZE = os.environ.get("MODIN_TEST_DATASET_SIZE", "Small")

BINARY_OP_DATA_SIZE = {
"Big": [
Expand Down

0 comments on commit 50f76c5

Please sign in to comment.