From e74d2525d39ff980561453ef6380aa1088570960 Mon Sep 17 00:00:00 2001
From: carlosuc3m <100329787@alumnos.uc3m.es>
Date: Tue, 5 Sep 2023 12:47:05 +0200
Subject: [PATCH] adapt to new jdll refacoring
---
.../deeplearningdownloader/DeepLearningVersionSelector.java | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningVersionSelector.java b/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningVersionSelector.java
index 0e1f4f6..45519be 100644
--- a/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningVersionSelector.java
+++ b/src/main/java/plugins/danyfel80/deeplearningdownloader/DeepLearningVersionSelector.java
@@ -48,11 +48,11 @@ public class DeepLearningVersionSelector extends EzPlug implements EzStoppable,
* @return map to let naive users select the wanted DL version
*/
private static Map<String, DeepLearningVersion> getDisplayMapWithoutPythonRepeated() {
- List<DeepLearningVersion> allVersions = AvailableEngines.getForCurrentOS().getVersions();
+ List<DeepLearningVersion> allVersions = AvailableEngines.getForCurrentOS();
List<DeepLearningVersion> vList = AvailableEngines.removeRepeatedPythonVersions(allVersions);
Map<String, DeepLearningVersion> map = vList.stream()
.collect(Collectors.toMap(
- v -> v.getEngine() + "-" + v.getPythonVersion()
+ v -> v.getFramework() + "-" + v.getPythonVersion()
+ (v.getCPU() ? "-" + DeepLearningVersion.cpuKey : "")
+ (v.getGPU() ? "-" + DeepLearningVersion.gpuKey : ""),
Function.identity()));
--
GitLab