From 03f9261df66abd5947a2117ffe494fa12d0cdae4 Mon Sep 17 00:00:00 2001
From: yancong xiao <32220263+ice-tong@users.noreply.github.com>
Date: Wed, 22 Jun 2022 20:09:51 +0800
Subject: [PATCH] Add more packages that support mim (#134)

* update(PKG2PROJECT): add more packages that support mim

* update(PKG2PROJECT): remove mmrazor and mmfewshot since the .mim folder is missing
---
 mim/utils/default.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mim/utils/default.py b/mim/utils/default.py
index f26c5a0..2e53299 100644
--- a/mim/utils/default.py
+++ b/mim/utils/default.py
@@ -23,6 +23,9 @@ PKG2PROJECT = {
     'mmedit': 'mmediting',
     'mmocr': 'mmocr',
     'mmgen': 'mmgeneration',
+    'mmselfsup': 'mmselfsup',
+    'mmrotate': 'mmrotate',
+    'mmflow': 'mmflow',
 }
 # TODO: Should directly infer MODULE name from PKG info
 PKG2MODULE = {