Patch distribution management bug

This commit is contained in:
YuigaWada
2023-09-30 16:45:00 +09:00
parent f533761b44
commit 2e44f9cdb3
2 changed files with 44 additions and 15 deletions
+42 -13
View File
@@ -279,18 +279,7 @@ const paramInternalValueToExternalValue = (
const parseDistributionJSON = (t: string): Distribution => {
const parsed = JSON.parse(t)
const floatDistributionList = [
"FloatDistribution",
"UniformDistribution",
"LogUniformDistribution",
"DiscreteUniformDistribution",
]
const intDistributionList = [
"IntDistribution",
"IntUniformDistribution",
"IntLogUniformDistribution",
]
if (floatDistributionList.includes(parsed.name)) {
if (parsed.name === "FloatDistribution") {
return {
type: "FloatDistribution",
low: parsed.attributes.low as number,
@@ -298,7 +287,31 @@ const parseDistributionJSON = (t: string): Distribution => {
step: parsed.attributes.step as number,
log: parsed.attributes.log as boolean,
}
} else if (intDistributionList.includes(parsed.name)) {
} else if (parsed.name === "UniformDistribution") {
return {
type: "FloatDistribution",
low: parsed.attributes.low as number,
high: parsed.attributes.high as number,
step: null,
log: false,
}
} else if (parsed.name === "LogUniformDistribution") {
return {
type: "FloatDistribution",
low: parsed.attributes.low as number,
high: parsed.attributes.high as number,
step: null,
log: true,
}
} else if (parsed.name === "DiscreteUniformDistribution") {
return {
type: "FloatDistribution",
low: parsed.attributes.low as number,
high: parsed.attributes.high as number,
step: parsed.attributes.q,
log: false,
}
} else if (parsed.name === "IntDistribution") {
return {
type: "IntDistribution",
low: parsed.attributes.low as number,
@@ -306,6 +319,22 @@ const parseDistributionJSON = (t: string): Distribution => {
step: parsed.attributes.step as number,
log: parsed.attributes.log as boolean,
}
} else if (parsed.name === "IntUniformDistribution") {
return {
type: "IntDistribution",
low: parsed.attributes.low as number,
high: parsed.attributes.high as number,
step: parsed.attributes.step as number,
log: false,
}
} else if (parsed.name === "IntLogUniformDistribution") {
return {
type: "IntDistribution",
low: parsed.attributes.low as number,
high: parsed.attributes.high as number,
step: parsed.attributes.step as number,
log: true,
}
} else {
// eslint-disable-next-line @typescript-eslint/no-explicit-any
const choices = parsed.attributes.choices.map((value: any) => {
+2 -2
View File
@@ -10,7 +10,7 @@ type FloatDistribution = {
type: "FloatDistribution"
low: number
high: number
step: number
step: number | null
log: boolean
}
@@ -18,7 +18,7 @@ type IntDistribution = {
type: "IntDistribution"
low: number
high: number
step: number
step: number | null
log: boolean
}