Properly handle vikings.

Author: GraylinKim

sc2reader/data/16939_units.csv

@@ -52,8 +52,8 @@
 16939|59|AutoTurret|AutoTurret
 16939|60|SiegeTankSieged|SiegeTankSieged
 16939|61|SiegeTank|SiegeTank
-16939|62|VikingAssault|VikingAssault
-16939|63|VikingFighter|VikingFighter
+16939|62|Viking|Viking
+16939|63|VikingAssault|VikingAssault
 16939|64|CommandCenterFlying|CommandCenterFlying
 16939|65|BarracksTechLab|BarracksTechLab
 16939|66|BarracksReactor|BarracksReactor

sc2reader/data/17811_units.csv

@@ -52,8 +52,8 @@
 17811|59|AutoTurret|AutoTurret
 17811|60|SiegeTankSieged|SiegeTankSieged
 17811|61|SiegeTank|SiegeTank
-17811|62|VikingAssault|VikingAssault
-17811|63|VikingFighter|VikingFighter
+17811|62|Viking|Viking
+17811|63|VikingAssault|VikingAssault
 17811|64|CommandCenterFlying|CommandCenterFlying
 17811|65|BarracksTechLab|BarracksTechLab
 17811|66|BarracksReactor|BarracksReactor

sc2reader/data/18701_units.csv

@@ -52,8 +52,8 @@
 18701|59|AutoTurret|AutoTurret
 18701|60|SiegeTankSieged|SiegeTankSieged
 18701|61|SiegeTank|SiegeTank
-18701|62|VikingAssault|VikingAssault
-18701|63|VikingFighter|VikingFighter
+18701|62|Viking|Viking
+18701|63|VikingAssault|VikingAssault
 18701|64|CommandCenterFlying|CommandCenterFlying
 18701|65|BarracksTechLab|BarracksTechLab
 18701|66|BarracksReactor|BarracksReactor

sc2reader/data/21029_units.csv

@@ -52,8 +52,8 @@
 21029|59|AutoTurret|AutoTurret
 21029|60|SiegeTankSieged|SiegeTankSieged
 21029|61|SiegeTank|SiegeTank
-21029|62|VikingAssault|VikingAssault
-21029|63|VikingFighter|VikingFighter
+21029|62|Viking|Viking
+21029|63|VikingAssault|VikingAssault
 21029|64|CommandCenterFlying|CommandCenterFlying
 21029|65|BarracksTechLab|BarracksTechLab
 21029|66|BarracksReactor|BarracksReactor

sc2reader/data/22612_units.csv

@@ -53,8 +53,8 @@
 22612|62|AutoTurret|AutoTurret
 22612|63|SiegeTankSieged|SiegeTankSieged
 22612|64|SiegeTank|SiegeTank
-22612|65|VikingAssault|VikingAssault
-22612|66|VikingFighter|VikingFighter
+22612|65|Viking|Viking
+22612|66|VikingAssault|VikingAssault
 22612|67|CommandCenterFlying|CommandCenterFlying
 22612|68|BarracksTechLab|BarracksTechLab
 22612|69|BarracksReactor|BarracksReactor

sc2reader/data/__init__.py

@@ -151,7 +151,7 @@ def __init__(self, build_id, units, abilities):
     'TrainBanshee': ('Banshee',60),
     'TrainRaven': ('Raven',60),
     'TrainBattlecruiser': ('Battlecruiser',90),
-    'TrainViking': ('VikingFighter',42),
+    'TrainViking': ('Viking',42),
     'MorphToPlanetaryFortress': ('PlanetaryFortress',50),
     'MorphToOrbitalCommand': ('OrbitalCommand',35),