chore: remove deprecate hive watermark model (#281)

* chore: remove deprecate hive watermark model * bump version
amundsen-io · Jun 4, 2020 · 06c1dbe · 06c1dbe
1 parent 29bfcea
commit 06c1dbe
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 135 deletions.
diff --git a/databuilder/models/hive_watermark.py b/databuilder/models/hive_watermark.py
diff --git a/example/dags/hive_sample_dag.py b/example/dags/hive_sample_dag.py
@@ -97,7 +97,7 @@ def create_table_wm_job(**kwargs):
     job_config = ConfigFactory.from_dict({
         'extractor.sqlalchemy.{}'.format(SQLAlchemyExtractor.CONN_STRING): connection_string(),
         'extractor.sqlalchemy.{}'.format(SQLAlchemyExtractor.EXTRACT_SQL): sql,
-        'extractor.sqlalchemy.model_class': 'databuilder.models.hive_watermark.HiveWatermark',
+        'extractor.sqlalchemy.model_class': 'databuilder.models.watermark.Watermark',
         'loader.filesystem_csv_neo4j.{}'.format(FsNeo4jCSVLoader.NODE_DIR_PATH):
             node_files_folder,
         'loader.filesystem_csv_neo4j.{}'.format(FsNeo4jCSVLoader.RELATION_DIR_PATH):

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 from setuptools import setup, find_packages
 
 
-__version__ = '2.5.19'
+__version__ = '2.6.0'
 
 requirements_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'requirements.txt')
 with open(requirements_path) as requirements_file:

diff --git a/tests/unit/models/test_hive_watermark.py b/tests/unit/models/test_hive_watermark.py