Fix zenml stack describe bug

src/zenml/cli/utils.py

@@ -181,9 +181,9 @@ def print_stack_configuration(
     )
     rich_table.add_column("COMPONENT_TYPE")
     rich_table.add_column("COMPONENT_NAME")
-    items = {typ.value: name for typ, name in config.items()}
+    items = ([typ.value, name] for typ, name in config.items())
     for item in items:
-        rich_table.add_row(*list(item))
+        rich_table.add_row(*item)
 
     # capitalize entries in first column
     rich_table.columns[0]._cells = [

src/zenml/steps/base_step.py

@@ -592,7 +592,7 @@ def __call__(
         )
 
         self.INPUT_SPEC = {
-            arg_name: artifact_type.type  # type:ignore[misc]
+            arg_name: artifact_type.type
             for arg_name, artifact_type in input_artifacts.items()
         }
 

tests/unit/cli/test_utils.py

@@ -18,6 +18,7 @@
 from hypothesis.strategies import datetimes
 
 from zenml.cli.utils import format_date, parse_unknown_options
+from zenml.repository import Repository
 
 SAMPLE_CUSTOM_ARGUMENTS = [
     '--custom_argument="value"',
@@ -41,3 +42,17 @@ def test_parse_unknown_options_returns_a_dict_of_known_options() -> None:
     assert isinstance(parsed_sample_args, dict)
     assert len(parsed_sample_args.values()) == 3
     assert parsed_sample_args["best_cat"] == '"aria"'
+
+
+def test_stack_config_has_right_contents_for_printing() -> None:
+    """Check that the stack config has the right components for printing"""
+    repo = Repository()
+    active_stack_name = repo.active_stack_name
+    stack_config = repo.stack_configurations[active_stack_name]
+    items = [[typ.value, name] for typ, name in stack_config.items()]
+    assert len(items) != 0
+    assert items is not None
+    for item in items:
+        assert isinstance(item, list)
+        for kv_pair in item:
+            assert isinstance(kv_pair, str)