From 3d96c857f3de9521f4d740f6be43082bcdb59dbf Mon Sep 17 00:00:00 2001
From: tharvik <tharvik@users.noreply.github.com>
Date: Wed, 7 Feb 2024 10:41:19 +0100
Subject: [PATCH] discojs-core/specs: prefer local import

---
 discojs/discojs-core/src/aggregator/mean.spec.ts       | 2 +-
 discojs/discojs-core/src/serialization/model.spec.ts   | 2 +-
 discojs/discojs-core/src/serialization/weights.spec.ts | 2 +-
 discojs/discojs-core/src/validation/validator.spec.ts  | 3 ++-
 4 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/discojs/discojs-core/src/aggregator/mean.spec.ts b/discojs/discojs-core/src/aggregator/mean.spec.ts
index fe1e6179e..23d9f2a31 100644
--- 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
index 147655edf..b544501d5 100644
--- 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
index eb2fdb7af..2a4a10b18 100644
--- 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
index 0e060a97c..d703f6ffb 100644
--- 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',