lib/earth/rail/rail_traction/data_miner.rb in earth-1.0.0 vs lib/earth/rail/rail_traction/data_miner.rb in earth-1.0.1

@@ -1,24 +1,18 @@
 require 'earth/rail/country_rail_traction'
 require 'earth/rail/country_rail_traction_class'
-require 'earth/rail/rail_company_traction'
-require 'earth/rail/rail_company_traction_class'
 
 RailTraction.class_eval do
   data_miner do
-    process "Ensure CountryRailTraction, CountryRailTractionClass, RailCompanyTraction, and RailCompanyTractionClass are populated" do
+    process "Ensure CountryRailTraction and CountryRailTractionClass are populated" do
       CountryRailTractionClass.run_data_miner!
       CountryRailTraction.run_data_miner!
-      RailCompanyTractionClass.run_data_miner!
-      RailCompanyTraction.run_data_miner!
     end
     
-    process "Derive rail traction names from CountryRailTraction, CountryRailTractionClass, RailCompanyTraction, and RailCompanyTractionClass" do
+    process "Derive rail traction names from CountryRailTraction and CountryRailTractionClass" do
       names = []
       names += CountryRailTractionClass.select("DISTINCT rail_traction_name").map(&:rail_traction_name)
       names += CountryRailTraction.select("DISTINCT rail_traction_name").map(&:rail_traction_name)
-      names += RailCompanyTractionClass.select("DISTINCT rail_traction_name").map(&:rail_traction_name)
-      names += RailCompanyTraction.select("DISTINCT rail_traction_name").map(&:rail_traction_name)
       names.uniq.each do |name|
         RailTraction.find_or_create_by_name(name)
       end
     end
   end