minor change (asdf warning)

BAMWelDX · Mar 6, 2024 · 89a294d · 89a294d
1 parent e758b18
commit 89a294d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/weldx/asdf/util.py b/weldx/asdf/util.py
@@ -10,9 +10,8 @@
 
 import asdf
 import pint
-from asdf.asdf import SerializationContext
 from asdf.config import AsdfConfig, get_config
-from asdf.extension import Extension
+from asdf.extension import Extension, SerializationContext
 from asdf.tagged import TaggedDict, TaggedList, TaggedString
 from asdf.util import uri_match as asdf_uri_match
 from boltons.iterutils import get_path, remap
@@ -494,7 +493,7 @@ def from_yaml_tree(
 
 def get_weldx_extension(ctx: SerializationContext | AsdfConfig) -> Extension:
     """Grab the weldx extension from list of current active extensions."""
-    if isinstance(ctx, asdf.asdf.SerializationContext):
+    if isinstance(ctx, asdf.extension.SerializationContext):
         extensions = ctx.extension_manager.extensions
     elif isinstance(ctx, asdf.config.AsdfConfig):
         extensions = ctx.extensions