replicate-client 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: efa5152767827d0b6d95d1b83cf211c9dfb82d11885401c74394d6dbd068949e
-  data.tar.gz: '040619b058d796d577bb7d1b75e21d84af4e1999b3ccebcc5b168ba5469241f6'
+  metadata.gz: 41640838ca41928bec457c4d1d15c6db9a72b04b8461d6b0827f57016c72bec3
+  data.tar.gz: 656d100154913dff9c7a91cda77a170fc8929eb1521fb2f67abda82191115712
 SHA512:
-  metadata.gz: 706867512328515a36e3af092bdd9bfb39a633e476cd37b7c4afc9f3e492be7f427c61eedc58e4ccaa85f151c5a4e17291bdcbbe1e7dc5a627c366c85dbf3125
-  data.tar.gz: 9a0c11660772e9f85f2e59676ce854cf9e2f4d79e2d8d701d6df6340bed17484a0297a13d1115eaed5d2b10a815be06d41feed24dce049ebdb7f2afd89ed6930
+  metadata.gz: dfee565e469de91d00a4ae6901995f4f7f311c6b9eec57f334c74fd67b15026168e6fd71b6e50f71487deb085eee6afb9369508f3511aa20395c8f321b77a5f1
+  data.tar.gz: f19e8a56c38d67c650acf91c787421b902b943b04cc645bae27851ee4801c245c0a2daead92517b6b60cdfaa9442f30d6484c265512720e97282c97ea60d7ea4

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    replicate-client (0.1.3)
+    replicate-client (0.1.4)
       faraday (>= 1)
 
 GEM

data/lib/replicate-client/training.rb

@@ -125,12 +125,12 @@ module ReplicateClient
     # The full model name in the format "owner/name".
     #
     # @return [String]
-    attr_accessor :model
+    attr_accessor :model_full_name
 
     # The version ID of the model being trained.
     #
     # @return [String]
-    attr_accessor :version
+    attr_accessor :version_id
 
     # The input data provided for the training.
     #
@@ -150,9 +150,14 @@ module ReplicateClient
 
     # The timestamp when the training was completed.
     #
-    # @return [String, nil]
+    # @return [Time, nil]
     attr_accessor :completed_at
 
+    # The timestamp when the training was started.
+    #
+    # @return [Time, nil]
+    attr_accessor :started_at
+
     # The logs generated during the training process.
     #
     # @return [String]
@@ -173,6 +178,11 @@ module ReplicateClient
     # @return [Hash, nil]
     attr_accessor :output
 
+    # The metrics generated during the training process.
+    #
+    # @return [Hash, nil]
+    attr_accessor :metrics
+
     # Initialize a new training instance.
     #
     # @param attributes [Hash] The attributes of the training.
@@ -232,6 +242,20 @@ module ReplicateClient
       reset_attributes(attributes)
     end
 
+    # The model instance of the training.
+    #
+    # @return [ReplicateClient::Model]
+    def model
+      @model ||= ReplicateClient::Model.find(model_full_name, version_id: version_id)
+    end
+
+    # The version instance of the training.
+    #
+    # @return [ReplicateClient::Model::Version]
+    def version
+      @version ||= model.version
+    end
+
     private
 
     # Set the attributes of the training.
@@ -241,8 +265,8 @@ module ReplicateClient
     # @return [void]
     def reset_attributes(attributes)
       @id = attributes["id"]
-      @model = attributes["model"]
-      @version = attributes["version"]
+      @model_full_name = attributes["model"]
+      @version_id = attributes["version"]
       @input = attributes["input"]
       @status = attributes["status"]
       @created_at = attributes["created_at"]
@@ -251,6 +275,11 @@ module ReplicateClient
       @error = attributes["error"]
       @urls = attributes["urls"]
       @output = attributes["output"]
+      @started_at = attributes["started_at"]
+      @metrics = attributes["metrics"]
+
+      @model = nil
+      @version = nil
     end
   end
 end

data/lib/replicate-client/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ReplicateClient
-  VERSION = "0.1.3"
+  VERSION = "0.1.4"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: replicate-client
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Dylan Player