Firm accept request conditionally

controller_db.py

@@ -45,15 +45,15 @@ class ControllerDB:
         Firm['Code'] = Firm['Code'].astype('string')
         Firm.fillna(0, inplace=True)
         list_dct = []
-        # for _, row in Firm.iterrows():
-        #     code = row['Code']
-        #     row = row['1':]
-        #     for product_code in row.index[row == 1].to_list():
-        #         dct = {code: [product_code]}
-        #         list_dct.append(dct)
-        #     #     break
-        #     # break
-        list_dct = [{'140': ['1.4.5.1']}]
+        for _, row in Firm.iterrows():
+            code = row['Code']
+            row = row['1':]
+            for product_code in row.index[row == 1].to_list():
+                dct = {code: [product_code]}
+                list_dct.append(dct)
+                break
+            break
+        # list_dct = [{'140': ['1.4.5.1']}]
         for idx_exp, dct in enumerate(list_dct):
             self.add_experiment_1(idx_exp, self.dct_parameter['n_max_trial'],
                                   dct)

firm.py

@@ -123,11 +123,12 @@ class FirmAgent(ap.Agent):
             # print(product.code, [firm.name for firm in list_firm])
 
     def accept_request(self, down_firm, product):
+        # if self.model.nprandom.choice([True, False], p=[0.1, 0.9]):
         self.firm_network.graph.add_edges_from([
             (self.firm_network.positions[self],
-             self.firm_network.positions[down_firm], {
-                 'Product': product.code
-             })
+            self.firm_network.positions[down_firm], {
+                'Product': product.code
+            })
         ])
         self.dct_prod_capacity[product] -= 1
         self.dct_request_prod_from_firm[product].remove(down_firm)