From 88ec48e1c4d93ba9cd3aa186c068ef2aa4c27c56 Mon Sep 17 00:00:00 2001 From: adamhrv Date: Mon, 17 Dec 2018 01:37:31 +0100 Subject: fixing dataset procesosrs --- megapixels/commands/datasets/gen_filepath.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'megapixels/commands/datasets/gen_filepath.py') diff --git a/megapixels/commands/datasets/gen_filepath.py b/megapixels/commands/datasets/gen_filepath.py index e06fee6b..5db405c0 100644 --- a/megapixels/commands/datasets/gen_filepath.py +++ b/megapixels/commands/datasets/gen_filepath.py @@ -50,7 +50,7 @@ def cli(ctx, opt_fp_in, opt_fp_out, opt_data_store, opt_dataset, opt_slice, from tqdm import tqdm from glob import glob - from app.models import DataStore + from app.models.data_store import DataStore from app.utils import file_utils, im_utils data_store = DataStore(opt_data_store, opt_dataset) @@ -97,6 +97,6 @@ def cli(ctx, opt_fp_in, opt_fp_out, opt_data_store, opt_dataset, opt_slice, file_utils.mkdirs(fp_out) df_filepath = pd.DataFrame.from_dict(data) df_filepath = df_filepath.sort_values(by=['subdir'], ascending=True) - df_filepath = df_filepath.reset_index(drop=True) + df_filepath = df_filepath.reset_index() df_filepath.index.name = 'index' df_filepath.to_csv(fp_out) \ No newline at end of file -- cgit v1.2.3-70-g09d2