From 8756de95cab61a773ae8fd5b5646708c6b1b9332 Mon Sep 17 00:00:00 2001
From: a252jain <a252jain@uwaterloo.ca>
Date: Fri, 5 Apr 2024 10:25:51 -0400
Subject: [PATCH] name changes
---
Sources/SwiftNLPGenericLLMMacros/Macros.swift | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/Sources/SwiftNLPGenericLLMMacros/Macros.swift b/Sources/SwiftNLPGenericLLMMacros/Macros.swift
index 0d72c05a..dd7e5fb7 100644
--- a/Sources/SwiftNLPGenericLLMMacros/Macros.swift
+++ b/Sources/SwiftNLPGenericLLMMacros/Macros.swift
@@ -132,25 +132,25 @@ public enum LLMPredictionFunctions: DeclarationMacro {
ret +=
"""
public static func prediction(input: \(model_input_name)) throws -> \(model_output_name) {
- let model = try \(model_type_name)(contentsOf: URL(fileURLWithPath: \"Sources/SwiftNLP/Models/\(model_url)\");
+ let model = try \(model_type_name)(contentsOf: URL(fileURLWithPath: \"Sources/SwiftNLP/Models/\(model_url)\"));
return try model.prediction(input: input, options: MLPredictionOptions())
}
public static func prediction(input: \(model_input_name), options: MLPredictionOptions) throws -> \(model_output_name) {
- let model = try \(model_type_name)(contentsOf: URL(fileURLWithPath: \"Sources/SwiftNLP/Models/\(model_url)\");
+ let model = try \(model_type_name)(contentsOf: URL(fileURLWithPath: \"Sources/SwiftNLP/Models/\(model_url)\"));
let outFeatures: MLFeatureProvider = try model.prediction(input: input, options:options)
return \(model_output_name)(features: outFeatures)
}
@available(macOS 13.6, iOS 17.0, tvOS 17.0, watchOS 10.0, *)
public static func prediction(input: \(model_input_name), options: MLPredictionOptions = MLPredictionOptions()) async throws -> \(model_output_name) {
- let model = try \(model_type_name)(contentsOf: URL(fileURLWithPath: \"Sources/SwiftNLP/Models/\(model_url)\");
+ let model = try \(model_type_name)(contentsOf: URL(fileURLWithPath: \"Sources/SwiftNLP/Models/\(model_url)\"));
let outFeatures: MLFeatureProvider? = try await model.prediction(input: input, options:options)
return \(model_output_name)(features: outFeatures!)
}
public static func predictions(inputs: [\(model_input_name)], options: MLPredictionOptions = MLPredictionOptions()) throws -> [\(model_output_name)] {
- let model = try \(model_type_name)(contentsOf: URL(fileURLWithPath: \"Sources/SwiftNLP/Models/\(model_url)\");
+ let model = try \(model_type_name)(contentsOf: URL(fileURLWithPath: \"Sources/SwiftNLP/Models/\(model_url)\"));
let res = try model.predictions(inputs: inputs, options: options);
return res;
}
--
GitLab