From 32d87f68dd9ddce365accd714f02928b5b556054 Mon Sep 17 00:00:00 2001
From: Athanasius <github@miggy.org>
Date: Sat, 3 Apr 2021 12:47:17 +0100
Subject: [PATCH] collate.py: Fix references to companion_category_map

---
 collate.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/collate.py b/collate.py
index 591f073b..9e05458d 100755
--- a/collate.py
+++ b/collate.py
@@ -12,7 +12,7 @@ from traceback import print_exc
 
 import companion
 import outfitting
-from edmc_data import ship_name_map
+from edmc_data import companion_category_map, ship_name_map
 
 
 def __make_backup(file_name: str, suffix: str = '.bak') -> None:
@@ -54,13 +54,13 @@ def addcommodities(data):
         new = {
             'id'       : commodity['id'],
             'symbol'   : commodity['name'],
-            'category' : data.companion_category_map.get(commodity['categoryname']) or commodity['categoryname'],
+            'category' : companion_category_map.get(commodity['categoryname']) or commodity['categoryname'],
             'name'     : commodity.get('locName') or 'Limpets',
         }
 
         old = commodities.get(key)
 
-        if old and data.companion_category_map.get(commodity['categoryname'], True):
+        if old and companion_category_map.get(commodity['categoryname'], True):
             if new['symbol'] != old['symbol'] or new['name'] != old['name']:
                 raise ValueError('{}: {!r} != {!r}'.format(key, new, old))