Skip to content

Commit c6e6093

Browse files
authored
fix desktop training config Regex Patterns (#1516)
1 parent 1a668e5 commit c6e6093

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

  • client/platform/desktop/backend/native

client/platform/desktop/backend/native/common.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -449,8 +449,8 @@ function getFilesWithExtensions(dir: string, extensions: string[], fileList: str
449449
*/
450450
async function getTrainingConfigs(settings: Settings): Promise<TrainingConfigs> {
451451
const pipelinePath = npath.join(settings.viamePath, 'configs/pipelines');
452-
const defaultTrainingConfiguration = 'train_detector_default.viame_csv.conf';
453-
const allowedPatterns = /\.viame_csv\.conf$/;
452+
const defaultTrainingConfiguration = 'train_detector_default.conf';
453+
const allowedPatterns = /train_.*\.conf$/;
454454
const disallowedPatterns = /.*(_nf|\.continue)\.viame_csv\.conf$/;
455455
const allowedModelExtensions = ['.zip', '.pth', '.pt', '.py', '.weights', '.wt'];
456456
const exists = await fs.pathExists(pipelinePath);

0 commit comments

Comments
 (0)