From 49c31d2cb3827aec1f60831a9105e1efa0bdcbef Mon Sep 17 00:00:00 2001 From: Nick Date: Wed, 30 Oct 2019 07:31:13 -0600 Subject: [PATCH] update --- generator/ctrl/training_utils/make_forest_tf_records.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/generator/ctrl/training_utils/make_forest_tf_records.py b/generator/ctrl/training_utils/make_forest_tf_records.py index 11217b8..7f93ce9 100644 --- a/generator/ctrl/training_utils/make_forest_tf_records.py +++ b/generator/ctrl/training_utils/make_forest_tf_records.py @@ -11,6 +11,9 @@ import argparse import fastBPE import platform import json +import sys +# insert at 1, 0 is the script path (or '' in REPL) +sys.path.insert(1, '../../..') from story.utils import * def make_samples_helper(context, story_block, action_results, path, tree_id):