Files changed (2) hide show
  1. executor.py +4 -4
  2. requirements.txt +1 -1
executor.py CHANGED
@@ -5,9 +5,9 @@ import datetime
5
  from pathlib import Path
6
  import json
7
 
8
- from risk_atlas_nexus.blocks.inference import WMLInferenceEngine
9
- from risk_atlas_nexus.blocks.inference.params import WMLInferenceEngineParams
10
- from risk_atlas_nexus.library import RiskAtlasNexus
11
 
12
  from functools import lru_cache, wraps
13
  from dotenv import load_dotenv
@@ -15,7 +15,7 @@ from dotenv import load_dotenv
15
  load_dotenv(override=True)
16
 
17
  # Load the taxonomies
18
- ran = RiskAtlasNexus() # type: ignore
19
 
20
 
21
  def clear_previous_risks():
 
5
  from pathlib import Path
6
  import json
7
 
8
+ from ai_atlas_nexus.blocks.inference import WMLInferenceEngine
9
+ from ai_atlas_nexus.blocks.inference.params import WMLInferenceEngineParams
10
+ from ai_atlas_nexus.library import AIAtlasNexus
11
 
12
  from functools import lru_cache, wraps
13
  from dotenv import load_dotenv
 
15
  load_dotenv(override=True)
16
 
17
  # Load the taxonomies
18
+ ran = AIAtlasNexus() # type: ignore
19
 
20
 
21
  def clear_previous_risks():
requirements.txt CHANGED
@@ -3,4 +3,4 @@ pydantic==2.9.2
3
  linkml==1.8.6
4
  linkml_runtime==1.8.3
5
  ibm_watsonx_ai==1.2.8
6
- risk-atlas-nexus [wml] @ git+https://github.com/IBM/risk-atlas-nexus
 
3
  linkml==1.8.6
4
  linkml_runtime==1.8.3
5
  ibm_watsonx_ai==1.2.8
6
+ ai-atlas-nexus [wml] @ git+https://github.com/IBM/ai-atlas-nexus