Bihaqo / t3f

Tensor Train decomposition on TensorFlow
https://t3f.readthedocs.io/en/latest/index.html
MIT License
222 stars 55 forks source link

Line 141, add np.any() #141

Closed lokhande-vishnu closed 6 years ago

lokhande-vishnu commented 6 years ago

Line 141, add np.any(), otherwise the code throws an error when max_tt_ranks is an array

coveralls commented 6 years ago

Coverage Status

Coverage remained the same at 93.528% when pulling 7459eba9b64533fbe911a1af54b09012f2f0484d on lokhande-vishnu:develop into 6b614ba0f7aff2f37045df4659e29204f9f2dcef on Bihaqo:develop.

Bihaqo commented 6 years ago

Thank you! If have time, would be awesome to also add a test that covers this problem so that it doesn't reappear in the future (a test that decomposes a tensor and passes an array as max_tt_rank).