wandb 0.1.8 → 0.1.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 17b68d037ed6930865fae0bd24437e1c41d3a746208c69237bf2cd9da3ed2c4e
-  data.tar.gz: be8a03c3963951579dadd5397780cdd1d003f0b2e7e83f6a181b6dda491267de
+  metadata.gz: ed086eee277a0665a3281f93764c8042edf254c30bded00812f3f332085e0ece
+  data.tar.gz: 4c2e7443bbc914c38701eb3baa00f858f2e7a9d25afcef274c198404512681cc
 SHA512:
-  metadata.gz: e7e1edea11690670d9d1af572c99caea8d6985fa31bd3e6255602352081d794bc00690a95042bf301588b160952de808be7f242b4f25cf6c54a8d146e966ccdb
-  data.tar.gz: 1c3120cf18ba4ee4cc2cd3e26f61cccc296cc7059b823f0d8892a6b27571a7de120a8d3378151010793b42e99539e722c18a7aae363af465cbc53da26098b7a1
+  metadata.gz: 52c8315c8d711a06263cbd98d296b7fca5030ee1ed155e5603a607c04348a36115595cd2cfa9ea9b4a76547d3d31cce07f2835496bd26c78cbf6f3cb824253aa
+  data.tar.gz: caa96acc54494baf471d5b06dc74dfdde85fb7d54e35ebdfee485b3c02cbfca044c2c3793161b6eed2b0816b1e43a2568818d143e180a9af7697edfb7728b73c

data/lib/wandb/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Wandb
-  VERSION = "0.1.8"
+  VERSION = "0.1.12"
 end

data/lib/wandb/xgboost_callback.rb

@@ -20,7 +20,9 @@ module Wandb
       end
     end
 
-    attr_accessor :project_name, :api_key, :custom_loggers, :history, :sample
+    attr_accessor :project_name, :api_key, :custom_loggers, :history, :sample,
+                  :log_model, :log_feature_importance, :importance_type, :define_metric,
+                  :normalize_feature_importance
 
     def initialize(options = {})
       options = Opts.new(options)
@@ -35,6 +37,19 @@ module Wandb
       @custom_loggers = options.default(:custom_loggers, [])
     end
 
+    def as_json
+      {
+        log_model: @log_model,
+        log_feature_importance: @log_feature_importance,
+        importance_type: @importance_type,
+        define_metric: @define_metric,
+        normalize_feature_importance: @normalize_feature_importance,
+        sample: @sample,
+        project_name: @project_name,
+        callback_type: :wandb,
+      }
+    end
+
     def before_training(model)
       Wandb.login(api_key: api_key)
       Wandb.init(project: project_name)
@@ -42,10 +57,8 @@ module Wandb
       log_conf = {
         learning_rate: config.dig("learner", "gradient_booster", "tree_train_param", "learning_rate").to_f,
         max_depth: config.dig("learner", "gradient_booster", "tree_train_param", "max_depth").to_f,
-        n_estimators: model.num_boosted_rounds
+        n_estimators: model.num_boosted_rounds,
       }
-      Wandb.current_run.config = log_conf
-
       Wandb.log(log_conf)
       model
     end
@@ -65,7 +78,7 @@ module Wandb
 
       Wandb.log(
         "best_score" => model.best_score.to_f,
-        "best_iteration" => model.best_iteration.to_i
+        "best_iteration" => model.best_iteration.to_i,
       )
       finish
 
@@ -93,9 +106,6 @@ module Wandb
           full_metric_name = "#{split}-#{metric}"
           Wandb.log({ full_metric_name => epoch_value })
         end
-        @custom_loggers.each do |logger|
-          logger.call(model, epoch, history)
-        end
         Wandb.log("epoch" => epoch)
       end
       false
@@ -126,7 +136,7 @@ module Wandb
       fi_data = fi.map { |k, v| [k, v] }
 
       table = Wandb::Table.new(data: fi_data, columns: %w[Feature Importance])
-      bar_plot = Wandb::Plot.bar(table.table, "Feature", "Importance", title: "Feature Importance")
+      bar_plot = Wandb::Plot.bar(table.table, label: "Feature", value: "Importance", title: "Feature Importance")
       Wandb.log({ "Feature Importance" => bar_plot.__pyptr__ })
     end
 

data/lib/wandb.rb

@@ -48,6 +48,14 @@ module Wandb
       @current_run = Run.new(run)
     end
 
+    def latest_run=(run)
+      @latest_run = run
+    end
+
+    def latest_run
+      @latest_run
+    end
+
     # Get the current run
     attr_reader :current_run
 
@@ -73,6 +81,12 @@ module Wandb
     def plot
       Plot
     end
+
+    def run_url
+      raise "No active run. Call Wandb.init() first." unless @current_run
+
+      @current_run.url
+    end
   end
 
   # Run class
@@ -113,6 +127,11 @@ module Wandb
     def log_artifact(artifact)
       @run.log_artifact(artifact.__pyptr__)
     end
+
+    # Add this new method
+    def url
+      @run.get_url
+    end
   end
 
   # Artifact class
@@ -232,8 +251,14 @@ module Wandb
   # Plot class
   class Plot
     class << self
-      def bar(table, x_key, y_key, title: nil)
-        py_plot = Wandb.__pyptr__.plot.bar(table.__pyptr__, x_key, y_key, title: title)
+      def bar(table, label:, value:, title: nil, split_table: false)
+        py_plot = Wandb.__pyptr__.plot.bar(
+          table: table.__pyptr__,
+          label: label,
+          value: value,
+          title: title,
+          split_table: split_table,
+        )
         new(py_plot)
       end
 
@@ -258,7 +283,7 @@ module Wandb
     end
 
     def __pyptr__
-      @plot
+      @plot.__pyptr__
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: wandb
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.12
 platform: ruby
 authors:
 - Brett Shollenberger
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-20 00:00:00.000000000 Z
+date: 2024-12-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pycall