Module: Projects::Ml::ModelRegistryHelper

Defined in:
app/helpers/projects/ml/model_registry_helper.rb

Instance Method Summary collapse

Instance Method Details

#edit_ml_model_data(model, user) ⇒ Object 



52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
 
# File 'app/helpers/projects/ml/model_registry_helper.rb', line 52

def edit_ml_model_data(model, user)
  project = model.project

  data = {
    projectPath: project.full_path,
    can_write_model_registry: can_write_model_registry?(user, project),
    markdown_preview_path: preview_markdown_path(project),
    model_path: project_ml_model_path(project, model),
    model_id: model.id,
    model_name: model.name,
    model_description: model.description.to_s
  }

  to_json(data)
end

#edit_ml_model_version_data(model_version, user) ⇒ Object 



104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
 
# File 'app/helpers/projects/ml/model_registry_helper.rb', line 104

def edit_ml_model_version_data(model_version, user)
  project = model_version.project

  data = {
    projectPath: project.full_path,
    can_write_model_registry: can_write_model_registry?(user, project),
    markdown_preview_path: preview_markdown_path(project),
    model_version_path: project_ml_model_version_path(project, model_version.model.id, model_version.id),
    model_gid: model_version.model.to_global_id.to_s,
    model_version_description: model_version.description.to_s,
    model_version_version: model_version.version
  }

  to_json(data)
end

#index_ml_model_data(project, user) ⇒ Object 



8
9
10
11
12
13
14
15
16
17
18
19
 
# File 'app/helpers/projects/ml/model_registry_helper.rb', line 8

def index_ml_model_data(project, user)
  data = {
    projectPath: project.full_path,
    create_model_path: new_project_ml_model_path(project),
    can_write_model_registry: can_write_model_registry?(user, project),
    mlflow_tracking_url: mlflow_tracking_url(project),
    max_allowed_file_size: max_allowed_file_size(project),
    markdown_preview_path: preview_markdown_path(project)
  }

  to_json(data)
end

#new_ml_model_data(project, user) ⇒ Object 



21
22
23
24
25
26
27
28
29
30
 
# File 'app/helpers/projects/ml/model_registry_helper.rb', line 21

def new_ml_model_data(project, user)
  data = {
    index_models_path: project_ml_models_path(project),
    projectPath: project.full_path,
    can_write_model_registry: can_write_model_registry?(user, project),
    markdown_preview_path: preview_markdown_path(project)
  }

  to_json(data)
end

#new_ml_model_version_data(model, user) ⇒ Object 



68
69
70
71
72
73
74
75
76
77
78
79
80
81
 
# File 'app/helpers/projects/ml/model_registry_helper.rb', line 68

def new_ml_model_version_data(model, user)
  project = model.project

  data = {
    model_path: project_ml_model_path(project, model),
    projectPath: project.full_path,
    can_write_model_registry: can_write_model_registry?(user, project),
    max_allowed_file_size: max_allowed_file_size(project),
    model_gid: model.to_global_id.to_s,
    markdown_preview_path: preview_markdown_path(project)
  }

  to_json(data)
end

#show_ml_model_data(model, user) ⇒ Object 



32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'app/helpers/projects/ml/model_registry_helper.rb', line 32

def show_ml_model_data(model, user)
  project = model.project

  data = {
    projectPath: project.full_path,
    index_models_path: project_ml_models_path(project),
    edit_model_path: edit_project_ml_model_path(project, model.id),
    create_model_version_path: new_project_ml_model_version_path(project, model.id),
    can_write_model_registry: can_write_model_registry?(user, project),
    mlflow_tracking_url: mlflow_tracking_url(project),
    model_id: model.id,
    model_name: model.name,
    max_allowed_file_size: max_allowed_file_size(project),
    latest_version: model.latest_version&.version,
    markdown_preview_path: preview_markdown_path(project)
  }

  to_json(data)
end

#show_ml_model_version_data(model_version, user) ⇒ Object 



83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
 
# File 'app/helpers/projects/ml/model_registry_helper.rb', line 83

def show_ml_model_version_data(model_version, user)
  project = model_version.project

  data = {
    project_path: project.full_path,
    model_id: model_version.model.id,
    model_version_id: model_version.id,
    model_name: model_version.name,
    version_name: model_version.version,
    can_write_model_registry: can_write_model_registry?(user, project),
    import_path: model_version_artifact_import_path(project.id, model_version.id),
    model_path: project_ml_model_path(project, model_version.model),
    edit_model_version_path: edit_project_ml_model_version_path(project, model_version.model.id,
      model_version.id),
    max_allowed_file_size: max_allowed_file_size(project),
    markdown_preview_path: preview_markdown_path(project)
  }

  to_json(data)
end