discojs-core/specs: prefer local import

epfml · Feb 7, 2024 · 3d96c85 · 3d96c85
1 parent aec8536
commit 3d96c85
Show file tree

Hide file tree

Showing 4 changed files with 5 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
@@ -1,7 +1,8 @@
 import { assert } from 'chai'
 import fs from 'fs'
 
-import { Task, node, Validator, ConsoleLogger, EmptyMemory, client as clients, data, aggregator } from '@epfml/discojs-node'
+import { Task, Validator, ConsoleLogger, EmptyMemory, client as clients, data, aggregator } from '..'
+import { node } from '@epfml/discojs-node'
 
 const simplefaceMock = {
   taskID: 'simple_face',