# coding: utf-8

from __future__ import division, print_function

__author__ = "adrn <adrn@astro.columbia.edu>"

# Standard library
import sys

# project
from ophiuchus.experiments import MockStreamGrid, ExperimentRunner

ExperimentRunner.parser.add_argument("--potential", dest="potential_name", required=True,
                                     help="Name of the potential YAML file.")

runner = ExperimentRunner(ExperimentClass=MockStreamGrid)
runner.run()

sys.exit(0)

@@ -125,16 +125,26 @@
 library%0A
+import os%0A
 import s
@@ -157,16 +157,50 @@
 project%0A
+from ophiuchus import RESULTSPATH%0A
 from oph
@@ -317,16 +317,17 @@
 , dest=%22
+_
 potentia
@@ -491,19 +491,567 @@
 id)%0A
-runner.run(
+%0A# Custom cache path%0Aargs = runner.parse_args()%0Aif args.results_path is None:%0A    results_path = RESULTSPATH%0Aelse:%0A    results_path = os.path.abspath(os.path.expanduser(args.results_path))%0A%0Aif results_path is None:%0A    raise ValueError(%22If $PROJECTSPATH is not set, you must provide a path to save %22%0A                     %22the results in with the --results_path argument.%22)%0A%0Aexperiment_name = runner.ExperimentClass.__name__.lower().rstrip(%22grid%22)%0Acache_path = os.path.join(results_path, args._potential_name, experiment_name)%0A%0Arunner.run(cache_path=cache_path
 )%0A%0As

need a few more lines