summaryrefslogtreecommitdiff
path: root/megapixels/commands/datasets/gen_filepath.py
diff options
context:
space:
mode:
authoradamhrv <adam@ahprojects.com>2018-12-17 01:37:31 +0100
committeradamhrv <adam@ahprojects.com>2018-12-17 01:37:31 +0100
commit88ec48e1c4d93ba9cd3aa186c068ef2aa4c27c56 (patch)
tree506075c0c8f0d4bbf15e97c6db50b6e055c5bd4e /megapixels/commands/datasets/gen_filepath.py
parent23e9fef5dce8b0b15dd94713816b9d7d45f12356 (diff)
fixing dataset procesosrs
Diffstat (limited to 'megapixels/commands/datasets/gen_filepath.py')
-rw-r--r--megapixels/commands/datasets/gen_filepath.py4
1 files changed, 2 insertions, 2 deletions
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