easy_ml 0.2.0.pre.rc51 → 0.2.0.pre.rc52

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 85c308aa34a6b29c804fc0e4c5eb44533ed046574f57563cbb0498148d159f2e
-  data.tar.gz: 16e352d22d97647027727a2b1964439f04e8b8e892472aa3ed4c900eba49e669
+  metadata.gz: b0b6da194500895ff325b1408757ee328b355da448a5fcc53439d7b14ccea9c8
+  data.tar.gz: da221df414245aafd56c5a05367146b9aae131885a99fa19e11aaa0f3cd32a20
 SHA512:
-  metadata.gz: 6fa1d47922112250c8d090105f37cab786742dd7e28872ba55565e08b9a1e6c67da8ac250254408874972adbca3e91c175edaa142effe9aaaaadd6620d9609a8
-  data.tar.gz: 737911077cc895c1b92f159e94f0cc8b565803c75e110b3ac4ffb70993427163e0fb7c52ec79b309f99a729dd00c5d9c60232ab1afc7addff264c641f9b6ee72
+  metadata.gz: 1d04422529423e09cff72496a29ff41038fdde5adfc0f267065981d04c24c9e9475d9ad0b2ab083e2d5e8cef22581180df97c93e960579cca8657b1771adc977
+  data.tar.gz: d9904acd55317dc68d774c26f392399990fdd52f45f394f8fdc6addf11456ea1361e2d972a438647c7dc0f679e3642cd8ae1f13cc9e0824c72865de8ecb0b6a9

data/app/frontend/components/ModelDetails.tsx

@@ -17,6 +17,16 @@ interface PaginatedRuns {
 
 const ITEMS_PER_PAGE = 3;
 
+const METRIC_MAPPINGS: Record<string, string> = {
+  root_mean_squared_error: 'rmse',
+  mean_absolute_error: 'mae',
+  mean_squared_error: 'mse',
+  r2_score: 'r2',
+  accuracy_score: 'accuracy',
+  precision_score: 'precision',
+  recall_score: 'recall'
+};
+
 export function ModelDetails({ model, onBack, rootPath }: ModelDetailsProps) {
   const [activeTab, setActiveTab] = useState<'overview' | 'dataset'>('overview');
   const [runs, setRuns] = useState<RetrainingRun[]>(model.retraining_runs?.runs || []);
@@ -103,6 +113,10 @@ export function ModelDetails({ model, onBack, rootPath }: ModelDetailsProps) {
     return run.status === 'deployed';
   };
 
+  const formatMetricKey = (key: string) => {
+    return METRIC_MAPPINGS[key] || key;
+  };
+
   return (
     <div className="space-y-6">
       <div className="flex items-center justify-between">
@@ -271,7 +285,7 @@ export function ModelDetails({ model, onBack, rootPath }: ModelDetailsProps) {
                       ).map(([key, value]) => (
                         <div key={key} className="bg-gray-50 rounded-md p-3">
                           <div className="text-sm font-medium text-gray-500">
-                            {key}
+                            {formatMetricKey(key)}
                           </div>
                           <div className="mt-1 flex items-center gap-2">
                             <span className="text-lg font-semibold">

data/app/jobs/easy_ml/compute_feature_job.rb

@@ -5,18 +5,18 @@ module EasyML
     @queue = :easy_ml
 
     def self.perform(batch_id, options = {})
-      options.symbolize_keys!
-      feature_id = options.dig(:feature_id)
-      feature = EasyML::Feature.find(feature_id)
-      dataset = feature.dataset
-
-      # Check if any feature has failed before proceeding
-      if dataset.features.any? { |f| f.workflow_status == "failed" }
-        puts "Aborting feature computation due to previous feature failure"
-        return
-      end
-
       begin
+        options.symbolize_keys!
+        feature_id = options.dig(:feature_id)
+        feature = EasyML::Feature.find(feature_id)
+        dataset = feature.dataset
+
+        # Check if any feature has failed before proceeding
+        if dataset.features.any? { |f| f.workflow_status == "failed" }
+          puts "Aborting feature computation due to previous feature failure"
+          return
+        end
+
         feature.update(workflow_status: :analyzing) if feature.workflow_status == :ready
         feature.fit_batch(options.merge!(batch_id: batch_id))
       rescue => e

data/app/jobs/easy_ml/refresh_dataset_job.rb

@@ -1,16 +1,16 @@
 module EasyML
   class RefreshDatasetJob < ApplicationJob
     def perform(id)
-      dataset = EasyML::Dataset.find(id)
-      puts "Refreshing dataset #{dataset.name}"
-      puts "Needs refresh? #{dataset.needs_refresh?}"
-      unless dataset.needs_refresh?
-        dataset.update(workflow_status: :ready)
-      end
+      begin
+        dataset = EasyML::Dataset.find(id)
+        puts "Refreshing dataset #{dataset.name}"
+        puts "Needs refresh? #{dataset.needs_refresh?}"
+        unless dataset.needs_refresh?
+          dataset.update(workflow_status: :ready)
+        end
 
-      create_event(dataset, "started")
+        create_event(dataset, "started")
 
-      begin
         puts "Prepare! #{dataset.name}"
         dataset.prepare
         if dataset.features.needs_fit.any?
@@ -25,6 +25,7 @@ module EasyML
         if Rails.env.test?
           raise e
         end
+        dataset.update(workflow_status: :failed)
         handle_error(dataset, e)
       end
     end

data/app/models/easy_ml/feature.rb

@@ -156,7 +156,7 @@ module EasyML
         raise "Couldn't find primary key for feature #{feature_class}, check your feature class"
       end
 
-      dataset.raw.data(limit: 1, select: primary_key)[primary_key].to_a.flat_map { |h| h.respond_to?(:values) ? h.values : h }.all? do |value|
+      dataset.raw.data(limit: 1, select: primary_key, all_columns: true)[primary_key].to_a.flat_map { |h| h.respond_to?(:values) ? h.values : h }.all? do |value|
         case value
         when String then value.match?(/\A[-+]?\d+(\.\d+)?\z/)
         else

data/app/serializers/easy_ml/dataset_serializer.rb

@@ -69,8 +69,8 @@ module EasyML
       if !object.failed? || object.events.empty?
         nil
       else
-        last_event = object.events.order(id: :desc).limit(1).last
-        last_event&.stacktrace if last_event&.status == "failed"
+        last_event = object.events.where(status: :failed).order(id: :desc).limit(1).last
+        last_event&.stacktrace
       end
     end
   end

data/lib/easy_ml/version.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module EasyML
-  VERSION = "0.2.0-rc51"
+  VERSION = "0.2.0-rc52"
 
   module Version
   end

data/public/easy_ml/assets/.vite/manifest.json

@@ -1,6 +1,6 @@
 {
   "entrypoints/Application.tsx": {
-    "file": "assets/entrypoints/Application.tsx-D08fSV1N.js",
+    "file": "assets/entrypoints/Application.tsx-BWHK2hEi.js",
     "name": "entrypoints/Application.tsx",
     "src": "entrypoints/Application.tsx",
     "isEntry": true,