Skip to content

Commit c8bf774

Browse files
committed
fix: fix ci issue base on update to serializer
1 parent 92f765e commit c8bf774

2 files changed

Lines changed: 7 additions & 3 deletions

File tree

src/cluster/kmeans.test.ts

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,10 +47,12 @@ describe('KMeans', () => {
4747
tol: 0.0001,
4848
randomState: 0,
4949
nInit: 10,
50-
clusterCenters: [
50+
clusterCenters: {
51+
type: "Tensor",
52+
value: [
5153
[2.5, 1],
5254
[2.5, 4]
53-
]
55+
] }
5456
}
5557
const kmean = new KMeans({ nClusters: 2, randomState: 0 })
5658
kmean.fit(X)

src/impute/simpleImputer.test.ts

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,9 @@ describe('SimpleImputer', function () {
134134
name: 'simpleimputer',
135135
missingValues: null,
136136
strategy: 'mostFrequent',
137-
statistics: [4, 3]
137+
statistics: {
138+
type: "Tensor",
139+
value: [4, 3] }
138140
}
139141

140142
const returned = imputer.fitTransform(data)

0 commit comments

Comments
 (0)