From 3e4d53761458340806afbad369b07c3382f84077 Mon Sep 17 00:00:00 2001
From: Marc Romeyn <mromeijn@nvidia.com>
Date: Thu, 22 Aug 2024 17:18:35 +0200
Subject: [PATCH] Run formatting again

Signed-off-by: Marc Romeyn <mromeijn@nvidia.com>
---
 src/nemo_run/cli/cli_parser.py | 18 ++++++++++++++----
 src/nemo_run/config.py         |  3 +--
 2 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/src/nemo_run/cli/cli_parser.py b/src/nemo_run/cli/cli_parser.py
index 293bf57..0d8408a 100644
--- a/src/nemo_run/cli/cli_parser.py
+++ b/src/nemo_run/cli/cli_parser.py
@@ -22,8 +22,19 @@
 from enum import Enum
 from functools import lru_cache
 from pathlib import Path
-from typing import (Any, Callable, Dict, List, Literal, Optional, Type,
-                    TypeVar, Union, get_args, get_origin)
+from typing import (
+    Any,
+    Callable,
+    Dict,
+    List,
+    Literal,
+    Optional,
+    Type,
+    TypeVar,
+    Union,
+    get_args,
+    get_origin,
+)
 
 import fiddle as fdl
 
@@ -1113,8 +1124,7 @@ def parse_factory(parent: Type, arg_name: str, arg_type: Type, value: str) -> An
     """
     import catalogue
 
-    from nemo_run.config import (Partial, get_type_namespace,
-                                 get_underlying_types)
+    from nemo_run.config import Partial, get_type_namespace, get_underlying_types
 
     def _get_from_registry(val, annotation, name):
         if catalogue.check_exists(get_type_namespace(annotation), val):
diff --git a/src/nemo_run/config.py b/src/nemo_run/config.py
index 6f025ef..dbcd10b 100644
--- a/src/nemo_run/config.py
+++ b/src/nemo_run/config.py
@@ -24,8 +24,7 @@
 import typing
 from pathlib import Path
 from types import MappingProxyType
-from typing import (Any, Callable, Generic, Optional, Type, TypeVar, Union,
-                    get_args)
+from typing import Any, Callable, Generic, Optional, Type, TypeVar, Union, get_args
 
 import fiddle as fdl
 import fiddle._src.experimental.dataclasses as fdl_dc