Fix TC StartupExtendedTest errors (#3888)

neo4j-contrib · Jan 8, 2024 · 8319990 · 8319990
1 parent 63f4bf7
commit 8319990
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/extended/src/main/java/apoc/ml/aws/Bedrock.java b/extended/src/main/java/apoc/ml/aws/Bedrock.java
@@ -7,6 +7,7 @@
 import java.util.stream.Stream;
 
 import apoc.Description;
+import apoc.Extended;
 import apoc.result.MapResult;
 import apoc.util.JsonUtil;
 import apoc.util.Util;
@@ -21,7 +22,7 @@
 import static apoc.util.JsonUtil.OBJECT_MAPPER;
 import static apoc.ml.aws.BedrockInvokeResult.*;
 
-
+@Extended
 public class Bedrock {
     @Context
     public URLAccessChecker urlAccessChecker;

diff --git a/extended/src/main/java/apoc/ml/aws/SageMaker.java b/extended/src/main/java/apoc/ml/aws/SageMaker.java
@@ -1,6 +1,7 @@
 package apoc.ml.aws;
 
 import apoc.Description;
+import apoc.Extended;
 import apoc.result.MapResult;
 import apoc.util.JsonUtil;
 import apoc.util.Util;
@@ -21,6 +22,7 @@
 import static apoc.ml.aws.SageMakerConfig.ENDPOINT_NAME_KEY;
 import static apoc.util.JsonUtil.OBJECT_MAPPER;
 
+@Extended
 public class SageMaker {
 
     @Context