added required comments

Signed-off-by: sumana sree <[email protected]>
flyteorg · Oct 13, 2024 · 606b78d · 606b78d
1 parent 44f079d
commit 606b78d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/data_types_and_io/data_types_and_io/tensorflow_type.py b/examples/data_types_and_io/data_types_and_io/tensorflow_type.py
@@ -28,6 +28,7 @@ def training_workflow(x: tf.Tensor, y: tf.Tensor) -> float:
     return evaluate_model(model=model, x=x, y=y)
 
 
+# TFRecord Files
 @task
 def process_tfrecord(file: TFRecordFile) -> int:
     count = 0
@@ -41,6 +42,7 @@ def tfrecord_workflow(file: TFRecordFile) -> int:
     return process_tfrecord(file=file)
 
 
+# TFRecord Directories
 @task
 def process_tfrecords_dir(dir: TFRecordsDirectory) -> int:
     count = 0