Update docs and fix inits

HallLab · Oct 9, 2020 · d1a3b49 · d1a3b49
1 parent 5b8c7d7
commit d1a3b49
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 25 deletions.
diff --git a/docs/index.rst b/docs/index.rst
@@ -9,18 +9,23 @@ Documentation for pandas-genomics
 
 Pandas ExtensionArray for working with genomics data
 
-TODO:
-
-- [x] Create the GenotypeArray ExtensionArray
-  - [x] Pass all necessary ExtensionArray tests
-- [ ] IO Functions
-  - [x] Load data from Plink files
-  - [ ] Export to Plink files
-  - [ ] Load data from VCF files
-  - [ ] Export to VCF files
-- [ ] Add Useful Methods
-  - [ ] Additive encoding
-  - [ ] Get variant frequencies
+Done
+----
+
+ - Create the GenotypeArray ExtensionArray
+ - Pass all necessary ExtensionArray tests
+ - Load data from Plink files
+
+ToDo
+----
+
+- IO
+  - Export to Plink files
+  - Load data from VCF files
+  - Export to VCF files
+- Add Useful Methods
+  - Additive encoding
+  - Get variant frequencies
 
 .. toctree::
    :maxdepth: 2

diff --git a/pandas_genomics/__init__.py b/pandas_genomics/__init__.py
@@ -1,14 +1,2 @@
-from .scalars import Genotype, Variant
-from .arrays import GenotypeDtype, GenotypeArray
-import io
-
-__all__ = [
-    'Genotype',
-    'Variant',
-    'GenotypeDtype',
-    'GenotypeArray',
-    'io'
-]
-
 # Simple version tracking for now until Poetry has a solution
 __version__ = "v0.2.0"
diff --git a/tests/genotype_array/test_genotypearray.py b/tests/genotype_array/test_genotypearray.py
@@ -3,7 +3,8 @@
 from pandas.tests.extension import base
 import pytest
 
-from pandas_genomics import GenotypeDtype, GenotypeArray, Variant
+from pandas_genomics.arrays import GenotypeDtype, GenotypeArray
+from pandas_genomics.scalars import Variant
 
 random.seed(1855)