easy_ml 0.2.0.pre.rc50 → 0.2.0.pre.rc52

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9571100b7780c1c76d4305566045b890cde5b7a4a0b4ad8526f26d8a3c6337aa
-  data.tar.gz: 129ba88803cfa45fc6382599e0e8615977567c461d2de9b70318f43d45a496ba
+  metadata.gz: b0b6da194500895ff325b1408757ee328b355da448a5fcc53439d7b14ccea9c8
+  data.tar.gz: da221df414245aafd56c5a05367146b9aae131885a99fa19e11aaa0f3cd32a20
 SHA512:
-  metadata.gz: bb6d4c3b678e58f813a8cc1822a75812fdeb0d2ad65b8d67ecb600bdf2822e522a3cae04c067ad91d4894e6f60dcf3e72bbe52a4206ac7892dde3ad0da8f9a89
-  data.tar.gz: 856438e590cec6306d64c64f7314b81c3eee907fc760fa1fd2978a0fff13fa71df5bc483aae3ee24a5ea7403043f6258f16c985e7e1e334c1e2782822786c56d
+  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/frontend/components/dataset/ColumnFilters.tsx

@@ -54,7 +54,7 @@ export function ColumnFilters({
   };
 
   const columnsWithNulls = columns
-    .filter(col => col.statistics?.processed.null_count && col.statistics.processed.null_count > 0)
+    .filter(col => col.statistics?.processed?.null_count && col.statistics?.processed?.null_count > 0)
     .sort((a, b) => calculateNullPercentage(b) - calculateNullPercentage(a));
 
   const [currentPage, setCurrentPage] = useState(1);

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/column.rb

@@ -98,10 +98,6 @@ module EasyML
 
     private
 
-    def set_date_column_if_date_splitter
-      binding.pry
-    end
-
     def handle_date_column_change
       return unless saved_change_to_is_date_column? && is_date_column?
 

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-rc50"
+  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-u6L4jkNy.js",
+    "file": "assets/entrypoints/Application.tsx-BWHK2hEi.js",
     "name": "entrypoints/Application.tsx",
     "src": "entrypoints/Application.tsx",
     "isEntry": true,