From 9da63bb42dc1f46e8f0d6c3efee7e959c318fb75 Mon Sep 17 00:00:00 2001
From: Michael <m.langguth@fz-juelich.de>
Date: Sat, 13 Feb 2021 20:45:29 +0100
Subject: [PATCH] Reintroduce destination_dir-argument to
preprocess_data_step2.py for consistency.
---
.../main_scripts/main_preprocess_data_step2.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/video_prediction_tools/main_scripts/main_preprocess_data_step2.py b/video_prediction_tools/main_scripts/main_preprocess_data_step2.py
index d5a57b8d..51894163 100644
--- a/video_prediction_tools/main_scripts/main_preprocess_data_step2.py
+++ b/video_prediction_tools/main_scripts/main_preprocess_data_step2.py
@@ -18,6 +18,7 @@ import warnings
def main():
parser = argparse.ArgumentParser()
parser.add_argument("-input_dir", type=str)
+ parser.add_argument("-dest_dir", type=str)
parser.add_argument("-sequence_length", type=int, default=20)
parser.add_argument("-sequences_per_file", type=int, default=20)
args = parser.parse_args()
@@ -90,6 +91,7 @@ def main():
if year_rank in real_years_months:
# Initilial instance
ins2 = ERA5Pkl2Tfrecords(input_dir=args.input_dir,
+ dest_dir=args.dest_dir,
sequence_length = args.sequence_length,
sequences_per_file=args.sequences_per_file)
# create the tfrecords-files
--
GitLab