Solved by ptrblck in post #2 
              
              
                Are you getting an error when you execute the import statement or is this just a warning from your IDE? 
OneCycleLR was added ~11 months ago, so it should be in 1.5.1.