Skip to content

Commit

Permalink
fix(client): add delegate option to non-callback init
Browse files Browse the repository at this point in the history
  • Loading branch information
stakach committed May 8, 2023
1 parent 1540914 commit 6063eee
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion shard.yml
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
name: tensorflow_lite
version: 1.2.0
version: 1.2.1

development_dependencies:
ameba:
Expand Down
4 changes: 2 additions & 2 deletions src/tensorflow_lite/client.cr
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,8 @@ class TensorflowLite::Client
@interpreter = Interpreter.new(@model, @options)
end

def self.new(model : Bytes | Path | Model | String, threads : Int? = nil)
Client.new(model, threads) { |error_message| Log.warn { error_message } }
def self.new(model : Bytes | Path | Model | String, delegate : Delegate? = nil, threads : Int? = nil)
Client.new(model, delegate, threads) { |error_message| Log.warn { error_message } }
end

getter model : Model
Expand Down

0 comments on commit 6063eee

Please sign in to comment.