Skip to content

Commit

Permalink
change the alias
Browse files Browse the repository at this point in the history
  • Loading branch information
benmalef committed Nov 20, 2024
1 parent e883c33 commit 1b02226
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions GANDLF/utils/pydantic_config.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from pydantic import BaseModel, ConfigDict, Field
from pydantic import BaseModel, ConfigDict
from typing import Dict, Any, List, Optional
from enum import Enum
from GANDLF.models.modelBase import ModelBase
Expand Down Expand Up @@ -64,7 +64,7 @@ class Parameters(BaseModel):
loss_function: Union[str, Dict]
data_augmentation: dict # TODO: maybe is better to create a class
nested_training: dict # TODO: maybe is better to create a class
optimizer: Union[dict, str] = Field(alias="opt")
optimizer: Union[dict, str]
patch_sampler: Union[dict, str]
patch_size: Union[List[int], int]
clip_mode: Union[str, None]
Expand All @@ -75,3 +75,4 @@ class Parameters(BaseModel):
output_dir: Optional[str] = ""
problem_type: Optional[str] = None
differential_privacy: Optional[dict] = {}
opt:Union[dict, str] # TODO: maybe is better to use alia (optimizer)

0 comments on commit 1b02226

Please sign in to comment.