Skip to content

Commit

Permalink
First round reviewer comments
Browse files Browse the repository at this point in the history
  • Loading branch information
johnml1135 committed Feb 6, 2024
1 parent 77a7f9b commit 2c5ff85
Show file tree
Hide file tree
Showing 10 changed files with 24 additions and 24 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -211,7 +211,7 @@ public static IMachineBuilder AddHangfireJobServer(

builder.Services.AddHangfireServer(o =>
{
o.Queues = [.. queues];
o.Queues = queues.ToArray();
});
return builder;
}
Expand Down
11 changes: 11 additions & 0 deletions src/SIL.Machine.AspNetCore/Models/ModelDownloadUrl.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
using System;

namespace SIL.Machine.AspNetCore.Models
{
public class ModelDownloadUrl
{
public string Url { get; set; } = default!;
public int ModelRevision { get; set; } = default!;
public DateTime ExipiresAt { get; set; } = default!;
}
}
2 changes: 1 addition & 1 deletion src/SIL.Machine.AspNetCore/Models/TranslationEngine.cs
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ public class TranslationEngine : IEntity
public string EngineId { get; set; } = default!;
public string SourceLanguage { get; set; } = default!;
public string TargetLanguage { get; set; } = default!;
public bool IsModelRetrievable { get; set; } = false;
public bool IsModelPersisted { get; set; } = false;
public int BuildRevision { get; set; }
public Build? CurrentBuild { get; set; }
}
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ Task CreateAsync(
string? engineName,
string sourceLanguage,
string targetLanguage,
bool isModelRetrievable = false,
bool isModelPersisted = false,
CancellationToken cancellationToken = default
);
Task DeleteAsync(string engineId, CancellationToken cancellationToken = default);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ public async Task<string> CreateJobScriptAsync(
+ $" 'shared_file_uri': '{baseUri}',\n"
+ $" 'shared_file_folder': '{folder}',\n"
+ (buildOptions is not null ? $" 'build_options': '''{buildOptions}''',\n" : "")
+ (engine.IsModelRetrievable ? $" 'save_model': '{engineId}_{engine.BuildRevision + 1}',\n" : "")
+ (engine.IsModelPersisted ? $" 'save_model': '{engineId}_{engine.BuildRevision + 1}',\n" : "")
+ $" 'clearml': True\n"
+ "}\n"
+ "run(args)\n";
Expand Down
8 changes: 4 additions & 4 deletions src/SIL.Machine.AspNetCore/Services/NmtEngineService.cs
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ public async Task CreateAsync(
string? engineName,
string sourceLanguage,
string targetLanguage,
bool isModelRetrievable = false,
bool isModelPersisted = false,
CancellationToken cancellationToken = default
)
{
Expand All @@ -47,7 +47,7 @@ await _engines.InsertAsync(
EngineId = engineId,
SourceLanguage = sourceLanguage,
TargetLanguage = targetLanguage,
IsModelRetrievable = isModelRetrievable
IsModelPersisted = isModelPersisted
},
cancellationToken
);
Expand Down Expand Up @@ -120,10 +120,10 @@ public async Task<ModelDownloadUrl> GetModelDownloadUrlAsync(
)
{
TranslationEngine engine = await GetEngineAsync(engineId, cancellationToken);
if (!engine.IsModelRetrievable)
if (!engine.IsModelPersisted)
throw new InvalidOperationException(
"The model cannot be downloaded. "
+ "To enable downloading the model, recreate the engine with IsModelRetrievable property to true."
+ "To enable downloading the model, recreate the engine with IsModelPersisted property to true."
);
if (engine.BuildRevision == 0)
throw new InvalidOperationException("The engine has not been built yet.");
Expand Down
2 changes: 1 addition & 1 deletion src/SIL.Machine.AspNetCore/Services/NmtTrainBuildJob.cs
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ await PipInstallModuleAsync(
+ $" 'shared_file_uri': '{_sharedFileService.GetBaseUri()}',\n"
+ (buildOptions is not null ? $" 'build_options': '''{buildOptions}''',\n" : "")
+ (
engine.IsModelRetrievable
engine.IsModelPersisted
? $" 'save_model': '{engine.Id}_{engine.BuildRevision + 1}',\n"
: ""
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ public async Task CreateAsync(
string? engineName,
string sourceLanguage,
string targetLanguage,
bool isModelRetrievable = false,
bool isModelPersisted = false,
CancellationToken cancellationToken = default
)
{
Expand All @@ -43,7 +43,7 @@ await _engines.InsertAsync(
EngineId = engineId,
SourceLanguage = sourceLanguage,
TargetLanguage = targetLanguage,
IsModelRetrievable = isModelRetrievable
IsModelPersisted = isModelPersisted
},
cancellationToken
);
Expand Down
11 changes: 0 additions & 11 deletions src/SIL.Machine/Translation/ModelDownloadUrl.cs

This file was deleted.

Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ private async Task SetUpAsync()
SourceLanguage = "es",
TargetLanguage = "en",
BuildRevision = 1,
IsModelRetrievable = true
IsModelPersisted = true
}
);
_engines.Add(
Expand All @@ -37,7 +37,7 @@ private async Task SetUpAsync()
SourceLanguage = "es",
TargetLanguage = "en",
BuildRevision = 2,
IsModelRetrievable = true
IsModelPersisted = true
}
);
async Task WriteFileStub(string path, string content)
Expand Down

0 comments on commit 2c5ff85

Please sign in to comment.