ecg gen: bug fix

alphanumericslab · Jul 11, 2024 · c7409bc · c7409bc
1 parent ce0d424
commit c7409bc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/codes/ecg-image-generator/ecg_plot.py b/codes/ecg-image-generator/ecg_plot.py
@@ -362,7 +362,7 @@ def ecg_plot(
         if columns > 1 and (i+1)%columns != 0:
             sep_x = [len(ecg[leadName])*step + x_offset + dc_offset + x_gap] * round(8*y_grid_dots)
             sep_x = np.array(sep_x)
-            sep_y = np.linspace(y_offset - 4*y_grid_dots*step, y_offset + 4*y_grid_dots*step, len(sep_x))
+            sep_y = np.linspace(y_offset - 4*y_grid_dots*0.002, y_offset + 4*y_grid_dots*0.002, len(sep_x))
             ax.plot(sep_x, sep_y, linewidth=line_width * 3, color=color_line)
 
     #Plotting longest lead for 12 seconds

diff --git a/codes/ecg-image-generator/gen_ecg_image_from_data.py b/codes/ecg-image-generator/gen_ecg_image_from_data.py
@@ -59,7 +59,7 @@ def get_parser():
     parser.add_argument('--random_bw',type=float,default=0)
     parser.add_argument('--remove_lead_names',action="store_false",default=True)
     parser.add_argument('--lead_name_bbox',action="store_true",default=False)
-    parser.add_argument('--store_config', type=int, const=1, default=0)
+    parser.add_argument('--store_config', type=int, nargs='?', const=1, default=0)
 
     parser.add_argument('--deterministic_offset',action="store_true",default=False)
     parser.add_argument('--deterministic_num_words',action="store_true",default=False)