Fix tf2_2d
dependency and remove workaround
#17
Merged
tf2_2d
dependency and remove workaround
#17