From 7f24b538dfd6189573f204e58c8effd218a12a41 Mon Sep 17 00:00:00 2001
From: "docs-update[bot]" <xsmahmed776@gmail.com>
Date: Thu, 3 Aug 2023 01:10:12 +0100
Subject: [PATCH] return OrderedPair from Jobresultbuilder class

---
 flojoy/job_result_builder.py | 4 ++--
 flojoy/small_memory.py       | 3 +--
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/flojoy/job_result_builder.py b/flojoy/job_result_builder.py
index 32b16dc..859c1a0 100644
--- a/flojoy/job_result_builder.py
+++ b/flojoy/job_result_builder.py
@@ -1,4 +1,4 @@
-from .data_container import DataContainer
+from .data_container import DataContainer, OrderedPair
 import numpy as np
 from typing import Union
 from .flojoy_instruction import FLOJOY_INSTRUCTION
@@ -67,4 +67,4 @@ def build(self):
     def get_default_data(self) -> DataContainer:
         x = np.arange(0, 1000, 1)  # type:ignore
         y = np.ones_like(x)
-        return DataContainer(x=x, y=y)
+        return OrderedPair(x=x, y=y)
diff --git a/flojoy/small_memory.py b/flojoy/small_memory.py
index 8190ec7..40156c6 100644
--- a/flojoy/small_memory.py
+++ b/flojoy/small_memory.py
@@ -1,4 +1,3 @@
-# import os, sys
 from typing import Any
 from .dao import Dao
 
@@ -61,7 +60,7 @@ def read_memory(self, job_id: str, key: str):
             case "dict":
                 return self.dao.get_obj(memory_key)
             case "np_array":
-                return self.dao.get_np_array(memory_key, meta_data)
+                return self.dao.get_np_array(memory_key)
             case "pd_dframe":
                 return self.dao.get_pd_dataframe(memory_key)
             case _: