fixed bug with only being able to run in main directory

studioml · Mar 8, 2019 · d3ef105 · d3ef105
1 parent 6cf3072
commit d3ef105
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tests/env_detect.py b/tests/env_detect.py
@@ -0,0 +1,20 @@
+import yaml
+
+AWSInstance = "aws" in list(
+    yaml.load(
+        open(
+            "./test_config.yaml",
+            "r"))["cloud"].keys())
+GCPInstance = "gcloud" in list(
+    yaml.load(
+        open(
+            "./test_config.yaml",
+            "r"))["cloud"].keys())
+
+
+def on_gcp():
+    return GCPInstance
+
+
+def on_aws():
+    return AWSInstance