easy_ml 0.2.0.pre.rc56 → 0.2.0.pre.rc58

data/lib/easy_ml/predict.rb

@@ -10,11 +10,17 @@ module EasyML
       @models = {}
     end
 
-    def self.predict(model_name, df, serialize: false)
+    def self.normalize_input(df)
       if df.is_a?(Hash)
         df = Polars::DataFrame.new(df)
       end
+      df
+    end
+
+    def self.predict(model_name, df, serialize: false)
+      df = normalize_input(df)
       raw_input = df.to_hashes
+
       df = instance.normalize(model_name, df)
       normalized_input = df.to_hashes
       preds = instance.predict(model_name, df)
@@ -52,6 +58,11 @@ module EasyML
       get_model(model_name).predict(df)
     end
 
+    def self.validate_input(model_name, df)
+      df = normalize_input(df)
+      instance.get_model(model_name).dataset.validate_input(df)
+    end
+
     def normalize(model_name, df)
       get_model(model_name).dataset.normalize(df, inference: true)
     end
@@ -72,7 +83,7 @@ module EasyML
     private
 
     def load_model(model_name)
-      current_model = EasyML::Model.find_by!(name: model_name).inference_version
+      current_model = EasyML::Model.find_by!(slug: model_name).inference_version
 
       # Load new model if not loaded or different version
       model_not_loaded = models[model_name].nil?

data/lib/easy_ml/railtie/generators/migration/migration_generator.rb

@@ -41,6 +41,9 @@ module EasyML
             add_workflow_status_to_easy_ml_features
             drop_path_from_easy_ml_model_files
             add_is_date_column_to_easy_ml_columns
+            add_computed_columns_to_easy_ml_columns
+            add_slug_to_easy_ml_models
+            add_default_to_is_target
           ].freeze
 
           # Specify the next migration number

data/lib/easy_ml/railtie/templates/migration/add_computed_columns_to_easy_ml_columns.rb.tt

@@ -0,0 +1,14 @@
+class AddComputedColumnsToEasyMLColumns < ActiveRecord::Migration[<%= ActiveRecord::Migration.current_version %>]
+  def change
+    add_column :easy_ml_columns, :computed_by, :string
+    add_column :easy_ml_columns, :is_computed, :boolean, default: false
+    
+    add_index :easy_ml_columns, :computed_by
+    add_index :easy_ml_columns, :is_computed
+
+    add_column :easy_ml_column_histories, :computed_by, :string
+    add_index :easy_ml_column_histories, :computed_by
+    add_column :easy_ml_column_histories, :is_computed, :boolean, default: false
+    add_index :easy_ml_column_histories, :is_computed
+  end
+end

data/lib/easy_ml/railtie/templates/migration/add_default_to_is_target.rb.tt

@@ -0,0 +1,6 @@
+class AddDefaultToIsTarget < ActiveRecord::Migration[<%= ActiveRecord::Migration.current_version %>]
+  def change
+    change_column_default(:easy_ml_columns, :is_target, false)
+    change_column_default(:easy_ml_column_histories, :is_target, false)
+  end
+end

data/lib/easy_ml/railtie/templates/migration/add_slug_to_easy_ml_models.rb.tt

@@ -0,0 +1,20 @@
+class AddSlugToEasyMLModels < ActiveRecord::Migration[<%= ActiveRecord::Migration.current_version %>]
+  def change
+    add_column :easy_ml_models, :slug, :string
+    add_index :easy_ml_models, :slug, unique: true
+
+    reversible do |dir|
+      dir.up do
+        execute <<-SQL
+          UPDATE easy_ml_models
+          SET slug = LOWER(REPLACE(name, ' ', '_'))
+        SQL
+      end
+    end
+
+    change_column_null :easy_ml_models, :slug, false
+
+    add_column :easy_ml_model_histories, :slug, :string
+    add_index :easy_ml_model_histories, :slug
+  end
+end

data/lib/easy_ml/version.rb

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

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

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