Bug 1317294 remove get records

moztelemetry/dataset.py

@@ -10,12 +10,12 @@
 from inspect import isfunction
 from itertools import chain, islice
 from multiprocessing import cpu_count
+
 from concurrent import futures
+from .heka import message_parser
 
-from moztelemetry import heka_message_parser
 from .store import S3Store
 
-
 MAX_CONCURRENCY = int(cpu_count() * 1.5)
 
 
@@ -194,7 +194,7 @@ def records(self, sc, limit=None, sample=1, decode=None, summaries=None):
                     .flatMap(lambda x:x)
 
         if decode is None:
-            decode = heka_message_parser.parse_heka_message
+            decode = message_parser.parse_heka_message
 
         return rdd.map(lambda x: self.store.get_key(x['key'])) \
                   .flatMap(lambda x: decode(x))

moztelemetry/heka/__init__.py

@@ -0,0 +1,3 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, you can obtain one at http://mozilla.org/MPL/2.0/.