Skip to content

Commit 0ddff57

Browse files
committed
debug ml guard init in connector executor
1 parent 14a403d commit 0ddff57

File tree

1 file changed

+2
-1
lines changed
  • ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote

1 file changed

+2
-1
lines changed

ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote/RemoteModel.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,8 @@ public void initModel(MLModel model, Map<String, Object> params, Encryptor encry
9292
this.connectorExecutor.setXContentRegistry((NamedXContentRegistry) params.get(XCONTENT_REGISTRY));
9393
this.connectorExecutor.setRateLimiter((TokenBucket) params.get(RATE_LIMITER));
9494
this.connectorExecutor.setUserRateLimiterMap((Map<String, TokenBucket>) params.get(USER_RATE_LIMITER_MAP));
95-
this.connectorExecutor.setMLGuard((MLGuard) params.get(GUARDRAILS));
95+
MLGuard mgd = (MLGuard) params.get(GUARDRAILS);
96+
this.connectorExecutor.setMLGuard(mgd);
9697
} catch (RuntimeException e) {
9798
log.error("Failed to init remote model.", e);
9899
throw e;

0 commit comments

Comments
 (0)