From 36517b7684e10cf9f24a0fb46d7d8be7e81a061a Mon Sep 17 00:00:00 2001
From: anmyachev <45976948+anmyachev@users.noreply.github.com>
Date: Tue, 28 Jul 2020 17:52:31 +0300
Subject: [PATCH] DOCS-#1835: add runner of taxi benchmark as example (#1836)

Signed-off-by: Anatoly Myachev <anatoly.myachev@intel.com>
---
 examples/cluster/taxi-runner.py | 52 +++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 examples/cluster/taxi-runner.py

diff --git a/examples/cluster/taxi-runner.py b/examples/cluster/taxi-runner.py
new file mode 100644
index 00000000000..a4e8128b24b
--- /dev/null
+++ b/examples/cluster/taxi-runner.py
@@ -0,0 +1,52 @@
+# Licensed to Modin Development Team under one or more contributor license agreements.
+# See the NOTICE file distributed with this work for additional information regarding
+# copyright ownership.  The Modin Development Team licenses this file to you under the
+# Apache License, Version 2.0 (the "License"); you may not use this file except in
+# compliance with the License.  You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software distributed under
+# the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific language
+# governing permissions and limitations under the License.
+
+
+# pip install git+https://github.com/intel-go/ibis.git@develop
+
+# NOTE: expects https://github.com/intel-go/omniscripts/ checked out and in PYTHONPATH
+
+# the following import turns on experimental mode in Modin,
+# including enabling running things in remote cloud
+import modin.experimental.pandas as pd  # noqa: F401
+from modin.experimental.cloud import create_cluster, get_connection
+
+from taxi import run_benchmark as run_benchmark
+from taxi.taxibench_pandas_ibis import etl_pandas
+
+test_cluster = create_cluster(
+    "aws",
+    "aws_credentials",
+    cluster_name="rayscale-test",
+    region="eu-north-1",
+    zone="eu-north-1b",
+    image="ami-00e1e82d7d4ca80d3",
+)
+with test_cluster:
+    conn = get_connection()
+    np = conn.modules["numpy"]
+    etl_pandas.__globals__["np"] = np
+
+    parameters = {
+        "data_file": "https://modin-datasets.s3.amazonaws.com/trips_data.csv",
+        # "data_file": "s3://modin-datasets/trips_data.csv",
+        "dfiles_num": 1,
+        "validation": False,
+        "no_ibis": True,
+        "no_pandas": False,
+        "pandas_mode": "Modin_on_ray",
+        "ray_tmpdir": "/tmp",
+        "ray_memory": 1024 * 1024 * 1024,
+    }
+
+    run_benchmark(parameters)