Update keras requirement from <3.0.0,>=2.0.0 to >=2.0.0,<4.0.0 in /tests

Update keras requirement from <3.0.0,>=2.0.0 to >=2.0.0,<4.0.0 in /tests #347