test: fix styling and imports

Author: wassimbensalem

sdk/python/kfp/compiler/pipeline_spec_builder.py

@@ -16,7 +16,8 @@
 import copy
 import json
 import typing
-from typing import Any, DefaultDict, Dict, List, Mapping, Optional, Tuple, Union
+from typing import (Any, DefaultDict, Dict, List, Mapping, Optional, Tuple,
+                    Union)
 import warnings
 
 from google.protobuf import json_format

sdk/python/kfp/dsl/pipeline_channel.py

@@ -322,10 +322,10 @@ class DictSubvariable(PipelineParameterChannel):
         def my_pipeline(config: dict):
             # Extract individual keys
             component1(param=config['db_host'])
-            
+
             # Extract nested values
             component2(host=config['database']['host'])
-            
+
             # Pass sub-dicts
             component3(db_config=config['database'])
 

sdk/python/kfp/dsl/pipeline_channel_test.py

@@ -430,7 +430,7 @@ def test_parent_must_be_pipeline_parameter_channel(self):
                 TypeError,
                 'parent_channel must be a PipelineParameterChannel'):
             pipeline_channel.DictSubvariable(
-                parent_channel="not_a_channel",
+                parent_channel='not_a_channel',
                 key='test_key'
             )