Merge pull request #19 from raff/python2

Quick and dirty fix for "bytes" (in torchaudio.save) not available in python2

test/test_transforms.py

+from __future__ import print_function
 import torch
 import torchaudio
 import torchaudio.transforms as transforms

torchaudio/__init__.py

 import os
+import sys
 
 import torch
 
@@ -10,6 +11,11 @@ from ._ext import th_sox
 from torchaudio import transforms
 from torchaudio import datasets
 
+if sys.version_info >= (3, 0):
+    _bytes = bytes
+else:
+    _bytes = lambda s, e: s.encode(e)
+
 def check_input(src):
     if not torch.is_tensor(src):
         raise TypeError('Expected a tensor, got %s' % type(src))
@@ -64,4 +70,4 @@ def save(filepath, src, sample_rate):
     check_input(src)
     typename = type(src).__name__.replace('Tensor', '')
     func = getattr(th_sox, 'libthsox_{}_write_audio_file'.format(typename))
-    func(bytes(filepath, "utf-8"), src, bytes(extension[1:], "utf-8"), sample_rate)
+    func(_bytes(filepath, "utf-8"), src, _bytes(extension[1:], "utf-8"), sample_rate)

torchaudio/transforms.py

-from __future__ import division
+from __future__ import division, print_function
 import torch
 import numpy as np
 try: