fix bugs in type hints (#77)

Co-authored-by: Sikai Yao <yaosk@dp.tech>

Author: MabinogiX

dpgen2/flow/dpgen_loop.py

@@ -14,6 +14,7 @@
     argo_len,
     argo_sequence,
     if_expression,
+    OPTemplate,
 )
 from dflow.python import(
     PythonOPTemplate,
@@ -114,7 +115,7 @@ class ConcurrentLearningLoop(Steps):
     def __init__(
             self,
             name : str,
-            block_op : Steps,
+            block_op : OPTemplate,
             step_config : dict = normalize_step_dict({}),
             upload_python_package : str = None,
     ):
@@ -201,7 +202,7 @@ class ConcurrentLearning(Steps):
     def __init__(
             self,
             name : str,
-            block_op : Steps,
+            block_op : OPTemplate,
             step_config : dict = normalize_step_dict({}),
             upload_python_package : str = None,
     ):
@@ -292,7 +293,7 @@ def _loop (
         steps, 
         step_keys,
         name : str,
-        block_op : OP,
+        block_op : OPTemplate,
         step_config : dict = normalize_step_dict({}),
         upload_python_package : str = None,
 ):    

dpgen2/superop/block.py

@@ -13,6 +13,7 @@
     argo_range,
     argo_len,
     argo_sequence,
+    OPTemplate,
 )
 from dflow.python import(
     PythonOPTemplate,
@@ -34,10 +35,10 @@ class ConcurrentLearningBlock(Steps):
     def __init__(
             self,
             name : str,
-            prep_run_dp_train_op : OP,
-            prep_run_lmp_op : OP,
+            prep_run_dp_train_op : OPTemplate,
+            prep_run_lmp_op : OPTemplate,
             select_confs_op : OP,
-            prep_run_fp_op : OP,
+            prep_run_fp_op : OPTemplate,
             collect_data_op : OP,
             select_confs_config : dict = normalize_step_dict({}),
             collect_data_config : dict = normalize_step_dict({}),
@@ -133,10 +134,10 @@ def _block_cl(
         block_steps : Steps,
         step_keys : List[str],
         name : str,
-        prep_run_dp_train_op : OP,
-        prep_run_lmp_op : OP,
+        prep_run_dp_train_op : OPTemplate,
+        prep_run_lmp_op : OPTemplate,
         select_confs_op : OP,
-        prep_run_fp_op : OP,
+        prep_run_fp_op : OPTemplate,
         collect_data_op : OP,
         select_confs_config : dict = normalize_step_dict({}),
         collect_data_config : dict = normalize_step_dict({}),