build_trains.rb in fastlane-2.74.0.beta.20180107010004

- old
+ new

@@ -1,5 +1,8 @@
+require_relative 'base'
+require_relative 'build'
+
 module Spaceship::TestFlight
   class BuildTrains < Base
     ##
     # BuildTrains represent the collection of builds for a `train_version`
     #
@@ -13,10 +16,10 @@
       data = client.get_build_trains(app_id: app_id, platform: platform)
       trains = {}
 
       data.each do |train_version|
         builds_data = client.get_builds_for_train(app_id: app_id, platform: platform, train_version: train_version, retry_count: retry_count)
-        trains[train_version] = builds_data.map { |attrs| Build.new(attrs) }
+        trains[train_version] = builds_data.map { |attrs| Spaceship::TestFlight::Build.new(attrs) }
       end
 
       self.new(trains)
     end
 