微修正（python）

https://github.com/microsoft/EconML/blob/master/notebooks/CustomerScenarios/Case%20Study%20-%20Recommendation%20AB%20Testing%20at%20An%20Online%20Travel%20Company.ipynb というマイクロソフトが公開しているコード（EconML）があるのですが、 1年前くらいに作成されたようで、現在は、モジュール変更されたとのことで、エラーが出てしまします。 こちらのコードが最後まで流れるように、修正してほしいです。変更箇所は多くないと思います。