discojs-core/specs: prefer local import

epfml · Feb 12, 2024 · 3a48d6c · 3a48d6c
1 parent a42b435
commit 3a48d6c
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/discojs/discojs-core/src/aggregator/mean.spec.ts b/discojs/discojs-core/src/aggregator/mean.spec.ts
@@ -2,7 +2,7 @@
 import { assert, expect } from 'chai'
 import { Map } from 'immutable'
 
-import { aggregator, defaultTasks, client, Task, tf } from '@epfml/discojs-node'
+import { aggregator, defaultTasks, client, Task, tf } from '..'
 import { AggregationStep } from './base'
 
 const task = defaultTasks.titanic.getTask()

diff --git a/discojs/discojs-core/src/serialization/model.spec.ts b/discojs/discojs-core/src/serialization/model.spec.ts
@@ -1,6 +1,6 @@
 import { assert } from 'chai'
 
-import { tf, serialization } from '@epfml/discojs-node'
+import { tf, serialization } from '..'
 
 async function getRawWeights (model: tf.LayersModel): Promise<Array<[number, Float32Array]>> {
   return Array.from(

diff --git a/discojs/discojs-core/src/serialization/weights.spec.ts b/discojs/discojs-core/src/serialization/weights.spec.ts
@@ -1,6 +1,6 @@
 import { assert } from 'chai'
 
-import { WeightsContainer, serialization } from '@epfml/discojs-node'
+import { WeightsContainer, serialization } from '..'
 
 describe('weights', () => {
   it('can encode what it decodes', async () => {

diff --git a/discojs/discojs-core/src/validation/validator.spec.ts b/discojs/discojs-core/src/validation/validator.spec.ts
@@ -32,7 +32,7 @@ describe('validator', () => {
         .map((file: string) => dir + subdir + file))
     const labels = files.flatMap((files, index) => Array(files.length).fill(index))
 
-    const data: data.Data = (await new node.data.NodeImageLoader(simplefaceMock)
+    const data = (await new node.data.NodeImageLoader(simplefaceMock)
       .loadAll(files.flat(), { labels })).train
     const buffer = new aggregator.MeanAggregator(simplefaceMock)
     const client = new clients.Local(new URL('http://localhost:8080'), simplefaceMock, buffer)