Fix some issues for #60

ceholden · Nov 20, 2015 · f574e41 · f574e41
1 parent a22f48a
commit f574e41
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/yatsm/cli/map.py b/yatsm/cli/map.py
@@ -11,17 +11,12 @@
                            get_coefficients, get_prediction)
 from yatsm.utils import write_output
 from yatsm.regression import design_coefs
-from yatsm.regression.transforms import harm
 
 gdal.AllRegister()
 gdal.UseExceptions()
 
 logger = logging.getLogger('yatsm')
 
-# number of days in year
-_days = 365.25
-w = 2 * np.pi / _days
-
 
 @click.command(short_help='Make map of YATSM output for a given date')
 @click.argument('map_type', metavar='<map_type>',

diff --git a/yatsm/mapping/prediction.py b/yatsm/mapping/prediction.py
@@ -8,6 +8,7 @@
 
 from .utils import find_result_attributes, find_indices
 from ..utils import find_results, iter_records
+from ..regression.transforms import harm
 
 logger = logging.getLogger('yatsm')
 
@@ -186,7 +187,7 @@ def get_prediction(date, result_location, image_ds,
                      dtype=np.int16) * int(ndv)
 
     logger.debug('Processing results')
-    for rec in iter_records(records, warn_on_empty=WARN_ON_EMPTY):
+    for rec in iter_records(records, warn_on_empty=warn_on_empty):
         for _qa, index in find_indices(rec, date, after=after, before=before):
             if index.shape[0] == 0:
                 continue