(#31)eden: tests: rename BaseBlock to Block

abraham-ai · Feb 5, 2022 · e2c10ea · e2c10ea
1 parent 21cd810
commit e2c10ea
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/eden/tests/server.py b/eden/tests/server.py
@@ -1,4 +1,4 @@
-from eden.block import BaseBlock
+from eden.block import Block
 from eden.datatypes import Image
 
 from eden.hosting import host_block
@@ -7,7 +7,7 @@
 ## set this to true if you want to run tests on gpus
 REQUIRES_GPU = False
 
-eden_block = BaseBlock(name = 'eden_block')
+eden_block = Block(name = 'eden_block')
 
 my_args = {
         'prompt': 'let there be tests', ## text

diff --git a/eden/tests/test_block.py b/eden/tests/test_block.py
@@ -1,7 +1,7 @@
 import unittest
 from unittest import TestCase
 
-from eden.block import BaseBlock
+from eden.block import Block
 from eden.datatypes import (
     Image,
 )
@@ -10,7 +10,7 @@ class TestBlock(unittest.TestCase):
 
     def test_block_init(self):
 
-        eden_block = BaseBlock(progress= True, name = 'some_block_name')
+        eden_block = Block(progress= True, name = 'some_block_name')
 
         self.assertTrue(eden_block.__run__ == None)
         self.assertTrue(eden_block.__setup__ == None)
@@ -21,7 +21,7 @@ def test_block_init(self):
 
     def test_block_build_pydantic_model(self):
 
-        eden_block = BaseBlock(progress= True, name = 'some_block_name')
+        eden_block = Block(progress= True, name = 'some_block_name')
 
         my_args = {
                 'prompt': 'let there be light', ## text