minor fix

Author: T-Dynamos

materialyoucolor/hct/__init__.py

@@ -0,0 +1 @@
+from .hct import Hct

materialyoucolor/utils/platform_utils.py

@@ -282,7 +282,7 @@ def logger(message):
     ):
         timer_start = default_timer()
         pixel_len = image.width * image.height
-        image_data = image.getdata()
+        image_data = image.get_flattened_data()
         # TODO: Think about getting data from bitmap
         pixel_array = [
             image_data[_]

tests/test_all.py

@@ -91,7 +91,7 @@ def get_current_rss_mb():
         start = default_timer()
         image = Image.open(args.image)
         pixel_len = image.width * image.height
-        image_data = image.getdata()
+        image_data = image.get_flattened_data()
         colors = QuantizeCelebi(
             [image_data[i] for i in range(0, pixel_len, args.quality)], MAX_COLOR
         )