Update src/bin/PROBE.py

src/bin/PROBE.py

@@ -51,7 +51,7 @@ def run_probe(benchmarks, representation_name, representation_file_human, repres
         results['family'] = {}
         for dataset in family_prediction_dataset:
             family_result = tfc.score_protein_rep(dataset)
-            results['family']['dataset'] = family_result
+            results['family'][f'{dataset}'] = family_result
 
     if "affinity" in benchmarks:
         print("\n\nProtein-protein binding affinity estimation benchmark is running...\n")