clean up

Arize-ai · Nov 1, 2024 · f18c45f · f18c45f
1 parent 9859cc8
commit f18c45f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...rence-instrumentation-guardrails/src/openinference/instrumentation/guardrails/__init__.py b/...rence-instrumentation-guardrails/src/openinference/instrumentation/guardrails/__init__.py
@@ -50,8 +50,6 @@ def instrumentation_dependencies(self) -> Collection[str]:
         return _instruments
 
     def _instrument(self, **kwargs: Any) -> None:
-        import guardrails as gd
-
         version = Version(metadata.version("guardrails-ai"))
         global GUARDRAILS_VERSION
         GUARDRAILS_VERSION = (version.major, version.minor, version.micro)
@@ -60,6 +58,8 @@ def _instrument(self, **kwargs: Any) -> None:
             logger.info("Guardrails version >= 0.5.2 detected, skipping instrumentation")
             return
 
+        import guardrails as gd
+
         if not (tracer_provider := kwargs.get("tracer_provider")):
             tracer_provider = trace_api.get_tracer_provider()
         if not (config := kwargs.get("config")):