Fix import

Ostorlab · Dec 4, 2024 · 1e6da0a · 1e6da0a
1 parent 5d62164
commit 1e6da0a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agent/subfinder_agent.py b/agent/subfinder_agent.py
@@ -2,7 +2,7 @@
 
 import logging
 
-from ruamel.yaml import YAML
+import ruamel.yaml
 from rich import logging as rich_logging
 import tld
 from ostorlab.agent import agent
@@ -31,7 +31,7 @@ def update_provider_config(
     config_path: str = CONFIG_PATH,
 ) -> None:
     """Update the Subfinder provider configuration file with the VirusTotal API key."""
-    yaml = YAML(typ="safe")
+    yaml = ruamel.yaml.YAML(typ="safe")
     yaml.default_flow_style = False  # Ensure block-style lists
     # Load existing configuration or initialize a new one
     try: